s3nh commited on
Commit
0e9a4d6
·
verified ·
1 Parent(s): 14e604f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -28,12 +28,11 @@ import whisper
28
 
29
 
30
  def load_whisper():
31
- return whisper.load_model("medium").to("cuda")
32
-
33
 
34
  def load_tts():
35
  tts_model = VitsModel.from_pretrained("facebook/mms-tts-pol")
36
- tts_model.to("cuda")
37
  tokenizer_tss = AutoTokenizer.from_pretrained("facebook/mms-tts-pol")
38
  return tts_model, tokenizer_tss
39
 
 
28
 
29
 
30
  def load_whisper():
31
+ return whisper.load_model("medium")
 
32
 
33
  def load_tts():
34
  tts_model = VitsModel.from_pretrained("facebook/mms-tts-pol")
35
+ #tts_model.to("cuda")
36
  tokenizer_tss = AutoTokenizer.from_pretrained("facebook/mms-tts-pol")
37
  return tts_model, tokenizer_tss
38