Update app.py

app.py

@@ -4,8 +4,8 @@ from transformers import ViTForImageClassification, ViTFeatureExtractor
 from PIL import Image
 
 # Load model and feature extractor
-model = ViTForImageClassification.from_pretrained('shahmi0519/banana_artificial', num_labels=2, ignore_mismatched_sizes=True)
-feature_extractor = ViTFeatureExtractor.from_pretrained('shahmi0519/banana_artificial')
+model = ViTForImageClassification.from_pretrained('shahmi0519/banana_artificial_v2', num_labels=2, ignore_mismatched_sizes=True)
+feature_extractor = ViTFeatureExtractor.from_pretrained('shahmi0519/banana_artificial_v2')
 
 # Move to GPU if available
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")