anonymousatom commited on
Commit
2b81b7a
1 Parent(s): 0bafdb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import BitsAndBytesConfig, pipeline
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):
12
  max_new_tokens = 200
 
6
 
7
 
8
  model_id = "marianna13/llava-phi-2-3b"
9
+ pipe = pipeline("image-to-text", model=model_id, trust_remote_code=True)
10
 
11
  def generate_text(image):
12
  max_new_tokens = 200