SRDdev commited on
Commit
0c25281
·
1 Parent(s): 3a01033

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,9 +5,9 @@ from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecode
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'
11
  feature_extractor = ViTFeatureExtractor.from_pretrained(encoder_checkpoint)
12
  tokenizer = AutoTokenizer.from_pretrained(decoder_checkpoint)
13
  model = VisionEncoderDecoderModel.from_pretrained(model_checkpoint).to(device)
 
5
 
6
 
7
  device = 'cpu'
8
+ encoder_checkpoint = 'google/vit-base-patch16-224-in21k'
9
+ decoder_checkpoint = 'distilgpt2'
10
+ model_checkpoint = '"gagan3012/ViTGPT2_vizwiz"'
11
  feature_extractor = ViTFeatureExtractor.from_pretrained(encoder_checkpoint)
12
  tokenizer = AutoTokenizer.from_pretrained(decoder_checkpoint)
13
  model = VisionEncoderDecoderModel.from_pretrained(model_checkpoint).to(device)