Update app.py

app.py

@@ -1,31 +1,26 @@
 import torch
-from transformers import VitsForConditionalGeneration, VitsTokenizer
+from transformers import pipeline
 import gradio as gr
 
-# ✅ Load model Yakut VITS (em thay model_id đúng nếu có)
-model_id = "facebook/mms-tts-sah"  # Yakut = sah
+# Tự động dùng GPU nếu có
+device = 0 if torch.cuda.is_available() else -1
 
-tokenizer = VitsTokenizer.from_pretrained(model_id)
-model = VitsForConditionalGeneration.from_pretrained(model_id)
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model.to(device)
+# Tải pipeline TTS cho tiếng Yakut (mã ngôn ngữ: sah)
+yakut_tts = pipeline("text-to-speech", model="facebook/mms-tts-sah", device=device)
 
-# ✅ Hàm chuyển văn bản Yakut thành audio
-def yakut_tts(text):
-    inputs = tokenizer(text, return_tensors="pt").to(device)
-    with torch.no_grad():
-        output = model.generate(**inputs)
-    audio = output.cpu().numpy()[0]
-    return (16000, audio)  # sample rate, waveform
+# Hàm chuyển văn bản sang giọng nói
+def yakut_voice(text):
+    result = yakut_tts(text)
+    return result["audio"]
 
-# ✅ Giao diện Gradio
-interface = gr.Interface(
-    fn=yakut_tts,
-    inputs=gr.Textbox(label="Nhập văn bản Yakut"),
-    outputs=gr.Audio(label="Phát âm Yakut", type="numpy"),
-    title="Yakut TTS App",
-    description="Ứng dụng chuyển văn bản tiếng Yakut sang giọng nói bằng VITS"
+# Giao diện Gradio
+app = gr.Interface(
+    fn=yakut_voice,
+    inputs=gr.Textbox(label="Nhập văn bản tiếng Yakut"),
+    outputs=gr.Audio(label="Giọng nói Yakut"),
+    title="Yakut Text-to-Speech",
+    description="Chuyển văn bản Yakut (sah) thành giọng nói bằng mô hình Facebook MMS"
 )
 
 if __name__ == "__main__":
-    interface.launch()
+    app.launch()