mirzaburanali commited on
Commit
7d564be
·
1 Parent(s): 00ad212

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = [["/content/test1.png"], ["/content/test2.png"],["/content/test3.png"]]
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,