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/fypvit', num_labels=30, ignore_mismatched_sizes=True)
-feature_extractor = ViTFeatureExtractor.from_pretrained('shahmi0519/fypvit')
+model = ViTForImageClassification.from_pretrained('Dhahlan2000/fruit_and_veg_chategorizer', num_labels=10, ignore_mismatched_sizes=True)
+feature_extractor = ViTFeatureExtractor.from_pretrained('Dhahlan2000/fruit_and_veg_chategorizer')
 
 # Move to GPU if available
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")