Update app.py

app.py

@@ -9,7 +9,7 @@ import logging
 logging.basicConfig(level=logging.DEBUG)
 
 # Load the pre-trained model and feature extractor
-model_name = "google/vit-base-patch16-224-in21k"
+model_name = "google/vit-base-patch16-224"
 logging.info("Loading image processor and model...")
 feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
 model = AutoModelForImageClassification.from_pretrained(model_name)