captain-awesome commited on
Commit
82bf579
·
verified ·
1 Parent(s): 2a13997

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -14,7 +14,7 @@ def caption_image(image):
14
  caption = processor.decode(out[0], skip_special_tokens=True)
15
  return caption
16
 
17
- image_input = gr.inputs.Image()
18
- caption_output = gr.outputs.Textbox()
19
 
20
- gr.Interface(caption_image, image_input, caption_output).launch()
 
 
 
14
  caption = processor.decode(out[0], skip_special_tokens=True)
15
  return caption
16
 
 
 
17
 
18
+
19
+ gr.Interface(caption_image, gr.inputs.Image(), "text").launch()
20
+ # gr.Interface(caption_image, image_input, caption_output).launch()