iapain commited on
Commit
3d1777f
·
1 Parent(s): 5650a80

fix: input

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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")