Download model from HF hub

app.py

@@ -3,8 +3,10 @@ import gc
 import os
 import sys
 import warnings
+from pathlib import Path
 
 import gradio as gr
+from huggingface_hub import snapshot_download
 
 warnings.filterwarnings("ignore")
 
@@ -39,8 +41,22 @@ def load_model():
         )
         cfg = WAN_CONFIGS["flf2v-14B"]
 
-        # Load from Hugging Face Hub
-        checkpoint_dir = MODEL_ID
+        # Download model from Hugging Face Hub to local cache
+        print("\nDownloading model files from Hugging Face Hub...", flush=True)
+        # Use HF_HOME environment variable if set (for Hugging Face Spaces)
+        # Otherwise use default cache location
+        cache_base = os.environ.get("HF_HOME")
+        if cache_base:
+            cache_dir_path = Path(cache_base) / "hub"
+        else:
+            cache_dir_path = Path.home() / ".cache" / "huggingface" / "hub"
+
+        checkpoint_dir = snapshot_download(
+            repo_id=MODEL_ID,
+            cache_dir=str(cache_dir_path),
+            local_files_only=False,
+        )
+        print(f"Model downloaded to: {checkpoint_dir}", flush=True)
 
         wan_flf2v_720P = wan.WanFLF2V(
             config=cfg,
@@ -305,9 +321,7 @@ def create_interface():
 
             with gr.Column(scale=1):
                 gr.Markdown("### 🎥 Generated Video")
-                result_video = gr.Video(
-                    label="Output Video", height=600, show_download_button=True
-                )
+                result_video = gr.Video(label="Output Video", height=600)
                 result_status = gr.Textbox(label="Status", interactive=False)
 
         # Event handlers