mrsu0994 commited on
Commit
ddfa4ca
·
1 Parent(s): 33a8940

upload f5-tts source

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -66,7 +66,7 @@ def run_f5_tts(ref_audio_path, ref_text, gen_text, model="F5TTS_Base", speed=1.2
66
 
67
  def generate_speech(ref_audio, ref_text, gen_text, speed, model):
68
  if ref_audio is None:
69
- return None, "⚠️ Vui lòng tải lên file audio tham chiếu!"
70
 
71
  audio_segment = AudioSegment.from_file(ref_audio)
72
  audio_segment = audio_segment.set_channels(1) # Chuyển sang mono
@@ -78,7 +78,10 @@ def generate_speech(ref_audio, ref_text, gen_text, speed, model):
78
  if os.path.exists(ref_audio_path):
79
  os.remove(ref_audio_path)
80
 
81
- return output_mp3, message
 
 
 
82
 
83
  # ====================== Gradio UI ======================
84
  with gr.Blocks() as interface:
@@ -95,11 +98,9 @@ with gr.Blocks() as interface:
95
  btn = gr.Button("🚀 Sinh giọng nói")
96
 
97
  with gr.Column():
98
- with gr.Group(): # gộp chung audio + trạng thái thành 1 khung
99
- output_audio = gr.Audio(type="filepath", label="🔊 Kết quả audio (.mp3)")
100
- output_status = gr.Textbox(label="📌 Trạng thái")
101
 
102
- btn.click(generate_speech, [ref_audio, ref_text, gen_text, speed, model], [output_audio, output_status])
103
 
104
  if __name__ == "__main__":
105
  interface.launch(server_name="0.0.0.0", server_port=7860)
 
66
 
67
  def generate_speech(ref_audio, ref_text, gen_text, speed, model):
68
  if ref_audio is None:
69
+ return None
70
 
71
  audio_segment = AudioSegment.from_file(ref_audio)
72
  audio_segment = audio_segment.set_channels(1) # Chuyển sang mono
 
78
  if os.path.exists(ref_audio_path):
79
  os.remove(ref_audio_path)
80
 
81
+ if output_mp3 is None:
82
+ raise Exception(message) # Hoặc xử lý lỗi theo cách khác, ví dụ print(message)
83
+
84
+ return output_mp3
85
 
86
  # ====================== Gradio UI ======================
87
  with gr.Blocks() as interface:
 
98
  btn = gr.Button("🚀 Sinh giọng nói")
99
 
100
  with gr.Column():
101
+ output_audio = gr.Audio(type="filepath", label="🔊 Kết quả audio (.mp3)")
 
 
102
 
103
+ btn.click(generate_speech, [ref_audio, ref_text, gen_text, speed, model], output_audio)
104
 
105
  if __name__ == "__main__":
106
  interface.launch(server_name="0.0.0.0", server_port=7860)