zanemotiwala commited on
Commit
0e3c47c
1 Parent(s): d538119

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -2,6 +2,10 @@ import gradio as gr
2
  from transformers import AutoTokenizer, VisionEncoderDecoderModel, ViTImageProcessor, ViTForImageClassification
3
  import torch
4
 
 
 
 
 
5
  # Initialize device and models for captioning
6
  device = 'cpu'
7
  encoder_checkpoint = "nlpconnect/vit-gpt2-image-captioning"
 
2
  from transformers import AutoTokenizer, VisionEncoderDecoderModel, ViTImageProcessor, ViTForImageClassification
3
  import torch
4
 
5
+ description = "Simple Image Captioning & Recognition App"
6
+ title = "This app allows users to upload an image to be transcribed into text and for classification. It can take a couple minutes for the image to text for high quality images."
7
+
8
+
9
  # Initialize device and models for captioning
10
  device = 'cpu'
11
  encoder_checkpoint = "nlpconnect/vit-gpt2-image-captioning"