download model from HF

webui.py

@@ -22,10 +22,16 @@ import gradio as gr
 from datetime import datetime
 from cli.SparkTTS import SparkTTS
 from sparktts.utils.token_parser import LEVELS_MAP_UI
+from huggingface_hub import snapshot_download
 
 
-def initialize_model(model_dir="pretrained_models/Spark-TTS-0.5B", device=0):
+def initialize_model(model_dir=None, device=0):
     """Load the model once at the beginning."""
+    
+    if model_dir is None:
+        logging.info(f"Downloading model to: {model_dir}")
+        model_dir = snapshot_download("SparkAudio/Spark-TTS-0.5B", local_dir="pretrained_models/Spark-TTS-0.5B")
+        
     logging.info(f"Loading model from: {model_dir}")
     device = torch.device(f"cuda:{device}")
     model = SparkTTS(model_dir, device)
@@ -213,7 +219,7 @@ def parse_arguments():
     parser.add_argument(
         "--model_dir",
         type=str,
-        default="pretrained_models/Spark-TTS-0.5B",
+        default=None,
         help="Path to the model directory."
     )
     parser.add_argument(
@@ -225,13 +231,13 @@ def parse_arguments():
     parser.add_argument(
         "--server_name",
         type=str,
-        default="0.0.0.0",
+        default=None,
         help="Server host/IP for Gradio app."
     )
     parser.add_argument(
         "--server_port",
         type=int,
-        default=7860,
+        default=None,
         help="Server port for Gradio app."
     )
     return parser.parse_args()