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 = ',-.:;'
|