ollui commited on
Commit
68de5cb
·
verified ·
1 Parent(s): 5e324ea

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -0
app.py ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
3
+ import soundfile as sf
4
+
5
+ # Tải model và processor
6
+ model = AutoModelForSpeechSeq2Seq.from_pretrained("facebook/mms-tts-mhr")
7
+ processor = AutoProcessor.from_pretrained("facebook/mms-tts-mhr")
8
+
9
+ # Nhập văn bản Meadow Mari
10
+ text = "Мыйым пычал!" # Ví dụ: "Chào buổi sáng!"
11
+
12
+ # Tiền xử lý
13
+ inputs = processor(text=text, return_tensors="pt")
14
+
15
+ # Sinh giọng nói
16
+ with torch.no_grad():
17
+ waveform = model.generate(**inputs)
18
+
19
+ # Ghi ra file âm thanh WAV
20
+ sf.write("mari_tts.wav", waveform.cpu().squeeze().numpy(), 16000)