zanemotiwala
commited on
Commit
•
0e3c47c
1
Parent(s):
d538119
Update app.py
Browse files
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"
|