Update app.py

ignore_mismatched_sizes=True

app.py

@@ -13,7 +13,9 @@ BASE_MODEL = "google/vit-base-patch16-224"
 def load_model():
     processor = AutoImageProcessor.from_pretrained(BASE_MODEL)
     model = ViTForImageClassification.from_pretrained(
-        BASE_MODEL, num_labels=3
+        BASE_MODEL,
+        num_labels=3,
+        ignore_mismatched_sizes=True
     )
     weights_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILE)
     state_dict = load_file(weights_path)