try to reload the model

predict.py

@@ -17,10 +17,7 @@ with open("labels.json", "r") as f:
 print("class_names:", class_names)
 
 # Load model
-
-model = SwinForImageClassification.from_pretrained("microsoft/swin-base-patch4-window7-224")
-
-model.classifier = torch.nn.Linear(model.classifier.in_features, len(class_names))
+model = SwinForImageClassification.from_pretrained("microsoft/swin-base-patch4-window7-224", num_labels=len(class_names))
 
 # Download the model file from the Hub if not present
 model_path = hf_hub_download(repo_id="Noha90/AML_16", filename="best_model.pth")