Upload folder using huggingface_hub

app.py

@@ -11,7 +11,7 @@ import torch
 import numpy as np  # 确保导入 numpy
 import random     # 确保导入 random
 import s3tokenizer
-
+from huggingface_hub import hf_hub_download
 from soulxpodcast.models.soulxpodcast import SoulXPodcast
 from soulxpodcast.config import Config, SoulXPodcastLLMConfig, SamplingParams
 from soulxpodcast.utils.dataloader import (
@@ -512,6 +512,10 @@ def render_interface() -> gr.Blocks:
     return page
 
 
+def prepare_model():
+    ckpt_path = hf_hub_download(repo_id="Soul-AILab/SoulX-Podcast-1.7B")
+    return ckpt_path
+
 # ================================================
 #                   Options
 # ================================================

requirements.txt

@@ -11,4 +11,5 @@ accelerate==1.10.1
 onnxruntime
 onnxruntime-gpu
 einops
-gradio
+gradio
+huggingface_hub