Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import ViTForImageClassification, ViTFeatureExtractor
 from PIL import Image
 
 # Load model and feature extractor
-model = ViTForImageClassification.from_pretrained('shahmi0519/mango_artificial', num_labels=10, ignore_mismatched_sizes=True)
+model = ViTForImageClassification.from_pretrained('shahmi0519/mango_artificial', num_labels=2, ignore_mismatched_sizes=True)
 feature_extractor = ViTFeatureExtractor.from_pretrained('shahmi0519/mango_artificial')
 
 # Move to GPU if available