Update

app.py

@@ -23,11 +23,11 @@ def _ensure_weights():
     """Download weights to cache if not present."""
     weights_path = hf_hub_download(
         repo_id="AlbeRota/UnReflectAnything", 
-        filename="weights/full_model_weights.pt"
+        filename="weights/full_model_weights.pt",
     )
     config_path = hf_hub_download(
         repo_id="AlbeRota/UnReflectAnything", 
-        filename="configs/pretrained_config.yaml"
+        filename="configs/pretrained_config.yaml",
     )
     return weights_path, config_path
 
@@ -80,8 +80,8 @@ def _get_model(device: str):
         pretrained=True,
         # weights_path=os.path.join(os.path.dirname(__file__), ".cache", "weights", "full_model_weights.pt"),
         # config_path=os.path.join(os.path.dirname(__file__), ".cache", "configs", "pretrained_config.yaml"),
-        weights_path=weights_path,
-        config_path=config_path,
+        weights_path="weights/full_model_weights.pt",
+        config_path="configs/pretrained_config.yaml",
         device=device,
         verbose=False,
     )
@@ -133,7 +133,8 @@ def build_ui():
             with gr.Column(scale=0, min_width=100):
                 LOGO_PATH = hf_hub_download(
                     repo_id="AlbeRota/UnReflectAnything", 
-                    filename="assets/logo.png"
+                    filename="assets/logo.png",
+                    cache_dir="assets"
                 )
                 if Path(LOGO_PATH).is_file():
                     gr.Image(