Update app.py

app.py

@@ -1,4 +1,4 @@
-import subprocess
+ import subprocess
 
 # Install necessary packages
 subprocess.run(["pip", "install", "-U", "git+https://github.com/huggingface/transformers.git"])
@@ -28,7 +28,7 @@ for i, label in enumerate(labels):
 
 # Load the pre-trained model and processor
 model_checkpoint = "microsoft/resnet-50"
-model = AutoModelForImageClassification.from_pretrained(model_checkpoint, num_labels=len(labels))
+model = AutoModelForImageClassification.from_pretrained(model_checkpoint, ignore_mismatched_sizes=True, num_labels=len(labels))
 image_processor = AutoImageProcessor.from_pretrained(model_checkpoint)
 
 # Define the image transformations
@@ -82,3 +82,4 @@ trainer = Trainer(
 # Evaluate the pre-trained model
 metrics = trainer.evaluate()
 print(metrics)
+