update gradio

gradio_app.py

@@ -11,6 +11,21 @@ from typing import List, Optional, Tuple, Union
 
 import gradio as gr
 import torch
+from huggingface_hub import snapshot_download
+
+def try_to_download_model():
+    repo_id = "tencent/HY-Motion-1.0"
+    target_folder = "HY-Motion-1.0-Lite"
+    print(f">>> start download ", repo_id, target_folder)
+    local_dir = snapshot_download(
+        repo_id=repo_id,
+        allow_patterns=f"{target_folder}/*",
+        local_dir="./downloaded_models"
+    )
+    final_model_path = os.path.join(local_dir, target_folder)
+    print(f">>> Final model path: {final_model_path}")
+    return final_model_path
+
 
 # Import spaces for Hugging Face Zero GPU support
 try:
@@ -60,7 +75,7 @@ def _init_runtime_if_needed():
     if "USE_HF_MODELS" not in os.environ:
         os.environ["USE_HF_MODELS"] = "1"
 
-    skip_text = True
+    skip_text = False
     _global_runtime = T2MRuntime(
         config_path=cfg,
         ckpt_name=ckpt,
@@ -801,12 +816,10 @@ class T2MGradioUI:
             )
 
 
-def create_demo():
+def create_demo(final_model_path):
     """Create the Gradio demo with Zero GPU support."""
     global _global_runtime, _global_args
 
-    final_model_path = './configs/base'
-
     class Args:
         model_path = final_model_path
         output_dir = "output/gradio"
@@ -857,7 +870,7 @@ def create_demo():
         if "USE_HF_MODELS" not in os.environ:
             os.environ["USE_HF_MODELS"] = "1"
 
-        skip_text = True
+        skip_text = False
         runtime = T2MRuntime(
             config_path=cfg,
             ckpt_name=ckpt,
@@ -874,7 +887,8 @@ def create_demo():
 
 
 # Create demo at module level for Hugging Face Spaces
-demo = create_demo()
+final_model_path = try_to_download_model()
+demo = create_demo(final_model_path)
 
 if __name__ == "__main__":
     demo.launch()