Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 = "
|
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 = [
|