Update app.py

app.py

@@ -3,6 +3,8 @@ from transformers import ViTForImageClassification, ViTFeatureExtractor, ViTConf
 import gradio as gr
 from PIL import Image
 
+
+print("p1")
 # Define the class labels as used during training
 labels = ['Leggings', 'Jogger', 'Palazzo', 'Cargo', 'Dresspants', 'Chinos']
 
@@ -12,7 +14,7 @@ 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")
-
+print("Model loaded")
 # Set the model to evaluation mode
 model.eval()