Update app.py
Browse files
app.py
CHANGED
@@ -12,8 +12,7 @@ tokenizer = VitsTokenizer.from_pretrained(model_name)
|
|
12 |
model = VitsModel.from_pretrained(model_name)
|
13 |
model.to("cuda")
|
14 |
|
15 |
-
tts_pipeline = pipeline("text-to-speech", model=model_name)
|
16 |
-
tts_pipeline.to("cuda")
|
17 |
|
18 |
new_sentence = '!.?'
|
19 |
in_sentence = ',-.:;'
|
|
|
12 |
model = VitsModel.from_pretrained(model_name)
|
13 |
model.to("cuda")
|
14 |
|
15 |
+
tts_pipeline = pipeline("text-to-speech", model=model_name, device="cuda")
|
|
|
16 |
|
17 |
new_sentence = '!.?'
|
18 |
in_sentence = ',-.:;'
|