Aspik101 commited on
Commit
394a7b9
Β·
1 Parent(s): 88f313e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,8 +24,8 @@ import whisper
24
  tts_model = VitsModel.from_pretrained("facebook/mms-tts-pol")
25
  tts_model.to("cuda")
26
  print("TTS Loaded!")
27
- tokenizer_tss = AutoTokenizer.from_pretrained("facebook/mms-tts-pol")
28
 
 
29
 
30
  def save_to_txt(text_to_save):
31
  with open('prompt.txt', 'w', encoding='utf-8') as f:
@@ -49,7 +49,7 @@ def _load_model_tokenizer():
49
  return model, tokenizer
50
 
51
 
52
-
53
  def postprocess(self, y):
54
  if y is None:
55
  return []
 
24
  tts_model = VitsModel.from_pretrained("facebook/mms-tts-pol")
25
  tts_model.to("cuda")
26
  print("TTS Loaded!")
 
27
 
28
+ tokenizer_tss = AutoTokenizer.from_pretrained("facebook/mms-tts-pol")
29
 
30
  def save_to_txt(text_to_save):
31
  with open('prompt.txt', 'w', encoding='utf-8') as f:
 
49
  return model, tokenizer
50
 
51
 
52
+ model, tokenizer = _load_model_tokenizer()
53
  def postprocess(self, y):
54
  if y is None:
55
  return []