Update app.py

app.py

@@ -4,8 +4,7 @@ import torch
 from huggingface_hub import hf_hub_download
 
 def inference(repo_id, model_name, img):
-    #model_path = hf_hub_download(repo_id=repo_id, filename=model_name)
-    model_path = 'trained_models/cFOS_in_HC/cFOS_in_HC_ensemble_1.pt'
+    model_path = hf_hub_download(repo_id=repo_id, filename=model_name)
     model = torch.jit.load(model_path, map_location='cpu')
     n_channels = len(model.norm.mean)