Fix downloaded file path

streamlit_apps/app_utils/depth_model.py

@@ -39,13 +39,16 @@ class DPTDepth(BaseDepthModel):
         weights_path = os.path.join("weights", weights_fname)
         if not os.path.isfile(weights_path):
             from huggingface_hub import hf_hub_download
-            hf_hub_download(repo_id="RGBD-SOD/S-MultiMAE", filename=weights_fname)
-            os.system(f"mv {weights_fname} weights")
+
+            downloaded_filepath = hf_hub_download(
+                repo_id="RGBD-SOD/S-MultiMAE", filename=weights_fname
+            )
+            os.system(f'mv "{downloaded_filepath}" weights')
         omnidata_ckpt = torch.load(
             weights_path,
             map_location="cpu",
         )
-        
+
         self.model = DPTDepthModel()
         self.model.load_state_dict(omnidata_ckpt)
         self.model: DPTDepthModel = self.model.to(device).eval()

streamlit_apps/app_utils/sod_selection_ui.py

@@ -29,14 +29,15 @@ def load_smultimae_model(
     cfg = arg_cfg[sod_model_config_key]()
 
     weights_fname = f"s-multimae-{cfg.experiment_name}-top{top}.pth"
-    ckpt_path = os.path.join(
-        "weights", weights_fname
-    )
+    ckpt_path = os.path.join("weights", weights_fname)
     print(ckpt_path)
     if not os.path.isfile(ckpt_path):
         from huggingface_hub import hf_hub_download
-        hf_hub_download(repo_id="RGBD-SOD/S-MultiMAE", filename=weights_fname)
-        os.system(f"mv {weights_fname} weights")
+
+        downloaded_filepath = hf_hub_download(
+            repo_id="RGBD-SOD/S-MultiMAE", filename=weights_fname
+        )
+        os.system(f'mv "{downloaded_filepath}" weights')
     assert os.path.isfile(ckpt_path)
 
     # sod_model = ModelPL.load_from_checkpoint(