Spaces:
Sleeping
Sleeping
Update app.py
Browse filesFix tokenizer variable name
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 =
|
27 |
with torch.no_grad():
|
28 |
-
waveform =
|
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 |
|