Update app.py

app.py

@@ -1,16 +1,18 @@
 import torch
-from transformers import ViTForImageClassification, ViTFeatureExtractor
+from transformers import ViTForImageClassification, ViTFeatureExtractor, ViTConfig
 import gradio as gr
 from PIL import Image
 
 # Define the class labels as used during training
 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")
+# Load the configuration
+config = ViTConfig.from_pretrained("DumbledoreWiz/PantsShape")
+
+# Load the ViT model with the configuration
 model = ViTForImageClassification.from_pretrained("DumbledoreWiz/PantsShape", config=config)
 feature_extractor = ViTFeatureExtractor.from_pretrained("DumbledoreWiz/PantsShape")
+
 # Set the model to evaluation mode
 model.eval()