Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -4,10 +4,15 @@ import torch
|
|
4 |
import soundfile as sf
|
5 |
|
6 |
|
|
|
7 |
# Initialize the VITS model, tokenizer, and seed
|
8 |
-
|
9 |
-
|
10 |
-
|
|
|
|
|
|
|
|
|
11 |
|
12 |
|
13 |
def text_to_speech(text_response):
|
|
|
4 |
import soundfile as sf
|
5 |
|
6 |
|
7 |
+
|
8 |
# Initialize the VITS model, tokenizer, and seed
|
9 |
+
|
10 |
+
vits_model = VitsModel.from_pretrained("espnet/kan-bayashi_ljspeech_vits")
|
11 |
+
vits_tokenizer = AutoTokenizer.from_pretrained("espnet/kan-bayashi_ljspeech_vits")
|
12 |
+
|
13 |
+
#vits_model = VitsModel.from_pretrained("facebook/mms-tts-eng")
|
14 |
+
#vits_tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-eng")
|
15 |
+
set_seed(555)
|
16 |
|
17 |
|
18 |
def text_to_speech(text_response):
|