Update app.py

app.py

@@ -1,20 +1,21 @@
+from transformers import VitsModel, VitsTokenizer
 import torch
-from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
 import soundfile as sf
 
-# Tải model và processor
-model = AutoModelForSpeechSeq2Seq.from_pretrained("facebook/mms-tts-mhr")
-processor = AutoProcessor.from_pretrained("facebook/mms-tts-mhr")
+# 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 Meadow Mari
-text = "Мыйым пычал!"  # Ví dụ: "Chào buổi sáng!"
+# Nhập văn bản muốn đọc
+text = "Привет"  # hoặc tiếng Meadow Mari nếu em có
 
-# Tiền xử lý
-inputs = processor(text=text, return_tensors="pt")
+# Token hóa
+inputs = tokenizer(text, return_tensors="pt")
 
-# Sinh giọng nói
+# Sinh audio
 with torch.no_grad():
-    waveform = model.generate(**inputs)
+    output = model(**inputs).waveform
 
-# Ghi ra file âm thanh WAV
-sf.write("mari_tts.wav", waveform.cpu().squeeze().numpy(), 16000)
+# Lưu file
+sf.write("output.wav", output.squeeze().numpy(), samplerate=22050)