Update app.py

app.py

@@ -27,8 +27,10 @@ for path in [model_path, config_path, preprocessor_path]:
         logging.info(f"Found file: {path}")
 
 # Load the model and feature extractor using the local directory
-feature_extractor = ViTFeatureExtractor.from_pretrained(model_dir)
-model = ViTForImageClassification.from_pretrained(model_dir)
+model_id = "google/vit-base-patch16-224"
+
+feature_extractor = ViTFeatureExtractor.from_pretrained(model_id)
+model = ViTForImageClassification.from_pretrained(model_path)
 
 # Ensure the model is in evaluation mode
 model.eval()