Spaces:
Sleeping
Sleeping
fix: input
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ from transformers import pipeline
|
|
| 4 |
model_id = "Salesforce/blip-image-captioning-large"
|
| 5 |
caption_eng = pipeline(model=model_id)
|
| 6 |
|
| 7 |
-
def evaluate():
|
| 8 |
return caption_eng(input)[0]["generated_text"]
|
| 9 |
|
| 10 |
iface = gr.Interface(fn=evaluate, inputs=gr.Image(type="pil"), outputs="text")
|
|
|
|
| 4 |
model_id = "Salesforce/blip-image-captioning-large"
|
| 5 |
caption_eng = pipeline(model=model_id)
|
| 6 |
|
| 7 |
+
def evaluate(input):
|
| 8 |
return caption_eng(input)[0]["generated_text"]
|
| 9 |
|
| 10 |
iface = gr.Interface(fn=evaluate, inputs=gr.Image(type="pil"), outputs="text")
|