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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -1,8 +1,5 @@
1
  # -*- coding: utf-8 -*-
2
 
3
- #pip install transformers -q
4
-
5
- #pip install gradio -q
6
 
7
  from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer
8
  import torch
@@ -33,7 +30,7 @@ outputs = [
33
  gr.outputs.Textbox(label = 'Caption')
34
  ]
35
 
36
- title = "Image Captioning with Visual Transformer using nlpconnect"
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 = [
 
1
  # -*- coding: utf-8 -*-
2
 
 
 
 
3
 
4
  from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer
5
  import torch
 
30
  gr.outputs.Textbox(label = 'Caption')
31
  ]
32
 
33
+ title = "Visual Transformer using nlpconnect for Image to Text generation"
34
  description = "ViT and GPT2 are used to generate Image Caption for the uploaded image. COCO Dataset was used for training."
35
  article = " <a href='https://huggingface.co/nlpconnect/vit-gpt2-image-captioning'>Model Repo on Hugging Face Model Hub</a>"
36
  examples = [