change model path

predict.py

@@ -30,7 +30,7 @@ class DeiT(nn.Module):
         return outputs.logits
 
 # Load model
-model_path = hf_hub_download(repo_id="Noha90/AML_16", filename="deit_best_model.pth", repo_type="space")
+model_path = hf_hub_download(repo_id="Noha90/AML_16", filename="deit_best_model.pth")
 print("Model path:", model_path)
 model = DeiT(num_classes=len(class_names))
 state_dict = torch.load(model_path, map_location="cpu")