Hugging Face's logo

Spaces:
gagan3012
/
ViTGPT2
Runtime error

App Files Community
gagan3012 commited on
Commit
6775ad2
·
1 Parent(s): fddfcea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ else:
11
 
12
  encoder_checkpoint = "google/vit-base-patch16-224-in21k"
13
  decoder_checkpoint = "distilgpt2"
14
- model_checkpoint = "gagan3012/ViTGPT2_vizwiz"
15
  feature_extractor = ViTFeatureExtractor.from_pretrained(encoder_checkpoint)
16
  tokenizer = AutoTokenizer.from_pretrained(decoder_checkpoint)
17
  model = VisionEncoderDecoderModel.from_pretrained(model_checkpoint).to(device)
 
11
 
12
  encoder_checkpoint = "google/vit-base-patch16-224-in21k"
13
  decoder_checkpoint = "distilgpt2"
14
+ model_checkpoint = "gagan3012/ViTGPT2I2A"
15
  feature_extractor = ViTFeatureExtractor.from_pretrained(encoder_checkpoint)
16
  tokenizer = AutoTokenizer.from_pretrained(decoder_checkpoint)
17
  model = VisionEncoderDecoderModel.from_pretrained(model_checkpoint).to(device)