OmBenz commited on
Commit
051d034
1 Parent(s): d9c8d1d

Update app.py

Browse files

Fix tokenizer variable name

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -13,7 +13,7 @@ asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base",
13
 
14
  # load text-to-speech checkpoint
15
  tts_tokenizer = VitsTokenizer.from_pretrained("facebook/mms-tts-fra")
16
- tts_model = VitsModel.from_pretrained("facebook/mms-tts-fra")
17
 
18
 
19
 
@@ -23,9 +23,9 @@ def translate(audio):
23
 
24
 
25
  def synthesise(text):
26
- inputs = tokenizer(text=text, return_tensors="pt").to(device)
27
  with torch.no_grad():
28
- waveform = model(**inputs).waveform[0]
29
  return waveform.cpu()
30
 
31
 
 
13
 
14
  # load text-to-speech checkpoint
15
  tts_tokenizer = VitsTokenizer.from_pretrained("facebook/mms-tts-fra")
16
+ tts_model = VitsModel.from_pretrained("facebook/mms-tts-fra").to(device)
17
 
18
 
19
 
 
23
 
24
 
25
  def synthesise(text):
26
+ inputs = tts_tokenizer(text=text, return_tensors="pt").to(device)
27
  with torch.no_grad():
28
+ waveform = tts_model(**inputs).waveform[0]
29
  return waveform.cpu()
30
 
31