Hugging Face's logo

Spaces:
Vidushee
/
AdLLaVA
Build error

App Files Community
1
Vidushee commited on
Commit
a7b729f
·
verified ·
1 Parent(s): be7703f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import BitsAndBytesConfig, pipeline
5
  # Load model directly
6
 
7
 
8
- model_id = "liuhaotian/llava-v1.5-13b-shard3gb"
9
  pipe = pipeline("image-to-text", model=model_id)
10
 
11
  def generate_text(image):
 
5
  # Load model directly
6
 
7
 
8
+ model_id = "marianna13/llava-phi-2-3b"
9
  pipe = pipeline("image-to-text", model=model_id)
10
 
11
  def generate_text(image):