Update model_utils.py

model_utils.py

@@ -27,8 +27,8 @@ class CheXNet(nn.Module):
 from huggingface_hub import hf_hub_download
 import torch
 
-def load_model(repo_id, filename, device):
-    model_path = hf_hub_download(repo_id=repo_id, filename=filename)
+def load_model(device):
+    model_path = hf_hub_download(repo_id="cfgpp/danny_net", filename="dannynet.pth")
     model = torch.load(model_path, map_location=device)
     model.eval()
     return model