Hugging Face's logo

Spaces:
gagan3012
/
ViTGPT2
Runtime error

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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ else:
10
  device = "cpu"
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)
 
10
  device = "cpu"
11
 
12
  encoder_checkpoint = "google/vit-base-patch16-224-in21k"
13
+ decoder_checkpoint = "gpt2"
14
  model_checkpoint = "gagan3012/ViTGPT2I2A"
15
  feature_extractor = ViTFeatureExtractor.from_pretrained(encoder_checkpoint)
16
  tokenizer = AutoTokenizer.from_pretrained(decoder_checkpoint)