fix model handling

app.py

@@ -8,6 +8,7 @@ import gradio as gr
 
 from transformers.utils import logging
 from transformers import set_seed
+from huggingface_hub import snapshot_download
 from model import VibeVoiceDemo
 
 logging.set_verbosity_info()
@@ -16,6 +17,8 @@ logger = logging.get_logger(__name__)
 
 
 
+snapshot_download(repo_id="microsoft/VibeVoice-1.5B", local_dir="./ckpts/vibevoice")
+
 def create_demo_interface(demo_instance: VibeVoiceDemo):
     """Create the Gradio interface with streaming support."""
     
@@ -574,7 +577,7 @@ def parse_args():
     parser.add_argument(
         "--model_path",
         type=str,
-        default="microsoft/VibeVoice-1.5B",
+        default="./ckpts/vibevoice",
         help="Path to the VibeVoice model directory",
     )
     parser.add_argument(
@@ -641,11 +644,11 @@ def main():
             max_size=20,  # Maximum queue size
             default_concurrency_limit=1  # Process one request at a time
         ).launch(
-            share=args.share,
-            # server_port=args.port,
-            server_name="0.0.0.0" if args.share else "127.0.0.1",
+            #share= args.share,
+            #server_port=args.port,
+            #server_name="0.0.0.0" if args.share else "127.0.0.1",
+            show_api=False, # Hide API docs for cleaner interface
             show_error=True,
-            show_api=False  # Hide API docs for cleaner interface
         )
     except KeyboardInterrupt:
         print("\n🛑 Shutting down gracefully...")

requirements.txt

@@ -1,2 +1,3 @@
 git+https://github.com/vibevoice-community/VibeVoice
 soundfile
+huggingface_hub