Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -67,7 +67,7 @@ def predict_step_pixel(dataset_pixel_values, model):
|
|
67 |
image methods
|
68 |
"""
|
69 |
def load_image2txt_model(image_model_name):
|
70 |
-
model = VisionEncoderDecoderModel.from_pretrained(image_model_name)
|
71 |
feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/swin-large-patch4-window7-224", use_auth_token=auth_token)
|
72 |
|
73 |
tokenizer = GPT2Tokenizer.from_pretrained("gpt2", use_auth_token=auth_token)
|
|
|
67 |
image methods
|
68 |
"""
|
69 |
def load_image2txt_model(image_model_name):
|
70 |
+
model = VisionEncoderDecoderModel.from_pretrained(image_model_name, local_files_only=True)
|
71 |
feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/swin-large-patch4-window7-224", use_auth_token=auth_token)
|
72 |
|
73 |
tokenizer = GPT2Tokenizer.from_pretrained("gpt2", use_auth_token=auth_token)
|