Update app.py

app.py

@@ -7,9 +7,10 @@ from PIL import Image
 labels = ['Leggings', 'Jogger', 'Palazzo', 'Cargo', 'Dresspants', 'Chinos']
 
 # Load the ViT model and feature extractor
-model = ViTForImageClassification.from_pretrained("DumbledoreWiz/PantsShape")
-feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224-in21k")
-
+#model = ViTForImageClassification.from_pretrained("DumbledoreWiz/PantsShape")
+#feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224-in21k")
+model = ViTForImageClassification.from_pretrained("DumbledoreWiz/PantsShape", config=config)
+feature_extractor = ViTFeatureExtractor.from_pretrained("DumbledoreWiz/PantsShape")
 # Set the model to evaluation mode
 model.eval()