Hugging Face's logo

Spaces:
team-indain-image-caption
/
Hindi-image-captioning
Runtime error

App Files Community
seanbenhur commited on
Commit
add49e0
•
1 Parent(s): 975ff93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import gradio as gr
4
  from transformers import GPT2Tokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel
5
 
6
 
7
-
8
  encoder_checkpoint = 'google/vit-base-patch16-224'
9
  decoder_checkpoint = 'surajp/gpt2-hindi'
10
  model_checkpoint = 'team-indain-image-caption/hindi-image-captioning'
 
4
  from transformers import GPT2Tokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel
5
 
6
 
7
+ device = 'cpu'
8
  encoder_checkpoint = 'google/vit-base-patch16-224'
9
  decoder_checkpoint = 'surajp/gpt2-hindi'
10
  model_checkpoint = 'team-indain-image-caption/hindi-image-captioning'