ruslanmv commited on
Commit
d9ee843
·
verified ·
1 Parent(s): 6f299f7

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -56,7 +56,7 @@ def generate_captions(image):
56
 
57
  return caption_git_base, caption_git_large, caption_blip_base, caption_blip_large, caption_vitgpt
58
 
59
- examples = [["cats.jpg"], ["stop_sign.png"], ["astronaut.jpg"]]
60
  outputs = [gr.outputs.Textbox(label="Caption generated by GIT-base"), gr.outputs.Textbox(label="Caption generated by GIT-large"), gr.outputs.Textbox(label="Caption generated by BLIP-base"), gr.outputs.Textbox(label="Caption generated by BLIP-large"), gr.outputs.Textbox(label="Caption generated by ViT+GPT-2")]
61
 
62
  title = "Interactive demo: comparing image captioning models"
 
56
 
57
  return caption_git_base, caption_git_large, caption_blip_base, caption_blip_large, caption_vitgpt
58
 
59
+ examples = [["cat.jpg"], ["dog.jpg"], ["horse.jpg"]]
60
  outputs = [gr.outputs.Textbox(label="Caption generated by GIT-base"), gr.outputs.Textbox(label="Caption generated by GIT-large"), gr.outputs.Textbox(label="Caption generated by BLIP-base"), gr.outputs.Textbox(label="Caption generated by BLIP-large"), gr.outputs.Textbox(label="Caption generated by ViT+GPT-2")]
61
 
62
  title = "Interactive demo: comparing image captioning models"