gdnartea commited on
Commit
f14d301
·
verified ·
1 Parent(s): 936c23e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
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
- vits_model = VitsModel.from_pretrained("facebook/mms-tts-eng")
9
- vits_tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-eng")
10
- set_seed(2020)
 
 
 
 
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):