Update app.py

app.py

@@ -13,7 +13,7 @@ from huggingface_hub import hf_hub_download
 
 # @title Data retrieval
 def download_weights():    
-    hf_hub_download(repo_id="mouseland/cellpose-sam", filename="cpsam")
+    return hf_hub_download(repo_id="mouseland/cellpose-sam", filename="cpsam")
     
     #os.system("wget -q https://huggingface.co/mouseland/cellpose-sam/resolve/main/cpsam")
 
@@ -42,8 +42,8 @@ def download_weights_old():
           fid.write(r.content)
 
 try:
-    download_weights()
-    model = models.CellposeModel(gpu=True , pretrained_model="cpsam")
+    fpath = download_weights()
+    model = models.CellposeModel(gpu=True , pretrained_model=fpath)
 except Exception as e:
     print(f"Error loading model: {e}")
     exit(1)