stazizov commited on
Commit
21e6214
1 Parent(s): a76b982

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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 = ',-.:;'