Update app.py

app.py

@@ -3,6 +3,9 @@ import torch
 from transformers import AutoFeatureExtractor, AutoTokenizer, VisionEncoderDecoderModel, GPT2Tokenizer, pipeline
 import os
 
+HF_DATASETS_OFFLINE=1
+TRANSFORMERS_OFFLINE=1
+
 device = 'cpu'
 auth_token = os.getenv("auth_token")
 #auth_token = os.environ.get("auth_token")
@@ -67,7 +70,7 @@ def predict_step_pixel(dataset_pixel_values, model):
     image methods
 """
 def load_image2txt_model(image_model_name):
-    model = VisionEncoderDecoderModel.from_pretrained(image_model_name, local_files_only=True)
+    model = VisionEncoderDecoderModel.from_pretrained(image_model_name)
     feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/swin-large-patch4-window7-224", use_auth_token=auth_token)
     
     tokenizer = GPT2Tokenizer.from_pretrained("gpt2", use_auth_token=auth_token)