fix: tensors

app.py

@@ -1,7 +1,7 @@
 import os
 import gradio as gr
 from scepter.modules.utils.file_system import FS
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, snapshot_download
 
 def resolve_hf_path(path):
     if isinstance(path, str) and path.startswith("hf://"):
@@ -17,7 +17,7 @@ def resolve_hf_path(path):
         if token is None:
             raise ValueError("HUGGINGFACE_HUB_TOKEN environment variable not set!")
         # If filename is provided, download that file; otherwise, download the whole repo snapshot.
-        local_path = hf_hub_download(repo_id=repo_id, filename=filename, token=token) if filename else hf_hub_download(repo_id=repo_id, token=token)
+        local_path = hf_hub_download(repo_id=repo_id, filename=filename, token=token) if filename else snapshot_download(repo_id=repo_id, token=token)
         return local_path
     return path