Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 = '
|
10 |
-
model_checkpoint = '
|
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)
|