Spaces:

team-indain-image-caption
/

Hindi-image-captioning

Runtime error

seanbenhur commited on Mar 27, 2022

Commit

975ff93

•

1 Parent(s): e90affb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ encoder_checkpoint = 'google/vit-base-patch16-224'
 decoder_checkpoint = 'surajp/gpt2-hindi'
 model_checkpoint = 'team-indain-image-caption/hindi-image-captioning'
 feature_extractor = ViTFeatureExtractor.from_pretrained(encoder_checkpoint)
-tokenizer = AutoTokenizer.from_pretrained(decoder_checkpoint)
 model = VisionEncoderDecoderModel.from_pretrained(model_checkpoint).to(device)

 decoder_checkpoint = 'surajp/gpt2-hindi'
 model_checkpoint = 'team-indain-image-caption/hindi-image-captioning'
 feature_extractor = ViTFeatureExtractor.from_pretrained(encoder_checkpoint)
+tokenizer = GPT2Tokenizer.from_pretrained(decoder_checkpoint)
 model = VisionEncoderDecoderModel.from_pretrained(model_checkpoint).to(device)