Spaces:
Build error
Build error
Update app.py
#1
by
anonymousatom
- opened
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
|