Spaces:
Paused
Paused
Update main.py
Browse files
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 = [["
|
| 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"
|