Commit
·
7d564be
1
Parent(s):
00ad212
Update app.py
Browse files
app.py
CHANGED
@@ -56,7 +56,7 @@ outputs = [gr.outputs.Textbox(label=f'Caption {i+1}') for i in range(num_return_
|
|
56 |
title = "Image Captioning using ViT + GPT2"
|
57 |
description = "Image caption is generated for the uploaded image using ViT and GPT2. For training, COCO Dataset was utilised. If you see any biases (gender, race, etc.) in our picture captioning model that we were unable to identify during our stress tests, please use the 'Flag' button to mark the offending image."
|
58 |
article = " <a href='https://huggingface.co/sachin/vit2distilgpt2'>Model Repo on Hugging Face Model Hub</a>"
|
59 |
-
examples = [["
|
60 |
|
61 |
gr.Interface(
|
62 |
predict_step,
|
|
|
56 |
title = "Image Captioning using ViT + GPT2"
|
57 |
description = "Image caption is generated for the uploaded image using ViT and GPT2. For training, COCO Dataset was utilised. If you see any biases (gender, race, etc.) in our picture captioning model that we were unable to identify during our stress tests, please use the 'Flag' button to mark the offending image."
|
58 |
article = " <a href='https://huggingface.co/sachin/vit2distilgpt2'>Model Repo on Hugging Face Model Hub</a>"
|
59 |
+
examples = [["test1.png"], ["test2.png"],["test3.png"]]
|
60 |
|
61 |
gr.Interface(
|
62 |
predict_step,
|