Inial commited on
Commit
8a15488
·
verified ·
1 Parent(s): 2df75ed

Create app.pu

Browse files
Files changed (1) hide show
  1. app.py +22 -0
app.py ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import torch
3
+ from transformers import AutoProcessor, VitsModel
4
+
5
+ model_id = "facebook/mms-tts-ava"
6
+
7
+ processor = AutoProcessor.from_pretrained(model_id)
8
+ model = VitsModel.from_pretrained(model_id)
9
+
10
+ def tts_fn(text):
11
+ inputs = processor(text=text, return_tensors="pt")
12
+ with torch.no_grad():
13
+ output = model(**inputs)
14
+ waveform = output.waveform.squeeze().numpy()
15
+ return (16000, waveform)
16
+
17
+ gr.Interface(
18
+ fn=tts_fn,
19
+ inputs=gr.Textbox(label="Nhập văn bản Avar"),
20
+ outputs=gr.Audio(label="Kết quả TTS", type="numpy"),
21
+ title="TTS Avar - facebook/mms-tts-ava"
22
+ ).launch()