VIZINTZOR commited on
Commit
16476a5
·
verified ·
1 Parent(s): 6b92f58

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +22 -0
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)