Update README.md
Browse files
README.md
CHANGED
@@ -6,3 +6,25 @@ tags:
|
|
6 |
- Thai
|
7 |
---
|
8 |
โมเดลนี้ใช้ เสียงที่บันทึกจาก Play.ht : https://play.ht/ เพื่อนำมา finetune model.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
- Thai
|
7 |
---
|
8 |
โมเดลนี้ใช้ เสียงที่บันทึกจาก Play.ht : https://play.ht/ เพื่อนำมา finetune model.
|
9 |
+
|
10 |
+
การใช้งาน :
|
11 |
+
|
12 |
+
import torch
|
13 |
+
from transformers import VitsTokenizer, VitsModel, set_seed
|
14 |
+
import scipy
|
15 |
+
|
16 |
+
tokenizer = VitsTokenizer.from_pretrained("VIZINTZOR/MMS-TTS-THAI-MALEV1",cache_dir="./mms")
|
17 |
+
model = VitsModel.from_pretrained("VIZINTZOR/MMS-TTS-THAI-MALEV1",cache_dir="./mms")
|
18 |
+
|
19 |
+
inputs = tokenizer(text="สวัสดีครับ นี่คือเสียงพูดภาษาไทย", return_tensors="pt")
|
20 |
+
|
21 |
+
set_seed(456) # make deterministic
|
22 |
+
|
23 |
+
with torch.no_grad():
|
24 |
+
outputs = model(**inputs)
|
25 |
+
|
26 |
+
waveform = outputs.waveform[0]
|
27 |
+
|
28 |
+
waveform_array = waveform.numpy()
|
29 |
+
|
30 |
+
scipy.io.wavfile.write("techno_output.wav", rate=model.config.sampling_rate, data=waveform_array)
|