try to reload the model

predict.py

@@ -25,12 +25,8 @@ model.classifier = torch.nn.Linear(model.classifier.in_features, 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")
 print("Model path:", model_path)
-state_dict = torch.load(model_path, map_location="cpu")
-
-# Remove incompatible keys (classifier weights)
-filtered_state_dict = {k: v for k, v in state_dict.items() if "classifier" not in k}
-model.load_state_dict(filtered_state_dict, strict=False)
 
+model.load_state_dict(torch.load((model_path)))
 model.eval()