parasmech commited on
Commit
2ec0341
1 Parent(s): e247b3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -9
app.py CHANGED
@@ -1,11 +1,4 @@
1
  # -*- coding: utf-8 -*-
2
- """VIT_demo_main.ipynb
3
-
4
- Automatically generated by Colaboratory.
5
-
6
- Original file is located at
7
- https://colab.research.google.com/drive/1h4UTeFydVldZcBzvaDxS0vtDgQA6rQ5K
8
- """
9
 
10
  #pip install transformers -q
11
 
@@ -44,9 +37,11 @@ title = "Image Captioning with Visual Transformer using nlpconnect"
44
  description = "ViT and GPT2 are used to generate Image Caption for the uploaded image. COCO Dataset was used for training."
45
  article = " <a href='https://huggingface.co/nlpconnect/vit-gpt2-image-captioning'>Model Repo on Hugging Face Model Hub</a>"
46
  examples = [
47
- ["Img_3.jpg"],
48
  ["Img_1.jpg"],
49
- ["Img_2.jpg"]
 
 
 
50
  ]
51
 
52
 
 
1
  # -*- coding: utf-8 -*-
 
 
 
 
 
 
 
2
 
3
  #pip install transformers -q
4
 
 
37
  description = "ViT and GPT2 are used to generate Image Caption for the uploaded image. COCO Dataset was used for training."
38
  article = " <a href='https://huggingface.co/nlpconnect/vit-gpt2-image-captioning'>Model Repo on Hugging Face Model Hub</a>"
39
  examples = [
 
40
  ["Img_1.jpg"],
41
+ ["Img_2.jpg"],
42
+ ["img_2t.jpg"],
43
+ ["img_t2.jpg"],
44
+ ["img4_t.jpg"]
45
  ]
46
 
47