Gilvan commited on
Commit
7ff0acf
1 Parent(s): cc1fd5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)