Update app.py

app.py

@@ -1,21 +1,27 @@
-from transformers import VitsModel, VitsTokenizer
+from transformers import VitsTokenizer, VitsModel
 import torch
 import soundfile as sf
+import gradio as gr
 
-# Tải tokenizer và model VITS
 model_id = "facebook/mms-tts-mhr"
 tokenizer = VitsTokenizer.from_pretrained(model_id)
 model = VitsModel.from_pretrained(model_id)
 
-# Nhập văn bản muốn đọc
-text = "Привет"  # hoặc tiếng Meadow Mari nếu em có
+def tts_mari(text):
+    inputs = tokenizer(text, return_tensors="pt")
+    with torch.no_grad():
+        output = model(**inputs)
+    audio = output.waveform.squeeze().cpu().numpy()
+    output_path = "output.wav"
+    sf.write(output_path, audio, 16000)
+    return output_path
 
-# Token hóa
-inputs = tokenizer(text, return_tensors="pt")
+interface = gr.Interface(
+    fn=tts_mari,
+    inputs=gr.Textbox(label="Nhập văn bản Meadow Mari"),
+    outputs=gr.Audio(type="filepath", label="Kết quả TTS"),
+    title="Meadow Mari TTS - Powered by Facebook VITS"
+)
 
-# Sinh audio
-with torch.no_grad():
-    output = model(**inputs).waveform
-
-# Lưu file
-sf.write("output.wav", output.squeeze().numpy(), samplerate=22050)
+if __name__ == "__main__":
+    interface.launch()