try to reload the model

predict.py

@@ -25,8 +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)
-
-model.load_state_dict(torch.load((model_path)))
+state_dict = torch.load(model_path, map_location="cpu")
+model.load_state_dict(state_dict)
 model.eval()