ZiyueJiang commited on
Commit
e98b81d
·
1 Parent(s): fb4669e

update head of readme

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -0
  2. tts/gradio_api.py +4 -0
requirements.txt CHANGED
@@ -15,3 +15,4 @@ torchdiffeq==0.2.5
15
  openai-whisper==20240930
16
  httpx==0.28.1
17
  gradio==5.23.1
 
 
15
  openai-whisper==20240930
16
  httpx==0.28.1
17
  gradio==5.23.1
18
+ huggingface_hub==0.29.3
tts/gradio_api.py CHANGED
@@ -20,6 +20,10 @@ import gradio as gr
20
  import traceback
21
  from tts.infer_cli import MegaTTS3DiTInfer, convert_to_wav, cut_wav
22
 
 
 
 
 
23
 
24
  def model_worker(input_queue, output_queue, device_id):
25
  device = None
 
20
  import traceback
21
  from tts.infer_cli import MegaTTS3DiTInfer, convert_to_wav, cut_wav
22
 
23
+ from huggingface_hub import hf_hub_download
24
+
25
+ hf_hub_download(repo_id="ByteDance/MegaTTS3", local_dir="./checkpoints")
26
+
27
 
28
  def model_worker(input_queue, output_queue, device_id):
29
  device = None