Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -82,7 +82,7 @@ def predict(
|
|
82 |
if task_name in ["S2TT", "T2TT"]:
|
83 |
tokens_ids = model.generate(**input_data, generate_speech=False, tgt_lang=target_language_code, num_beams=5, do_sample=True)[0].cpu().squeeze().detach().tolist()
|
84 |
else:
|
85 |
-
output = model.generate(**input_data, return_intermediate_token_ids=True, tgt_lang=target_language_code, num_beams=5, do_sample=True
|
86 |
|
87 |
waveform = output.waveform.cpu().squeeze().detach().numpy()
|
88 |
tokens_ids = output.sequences.cpu().squeeze().detach().tolist()
|
|
|
82 |
if task_name in ["S2TT", "T2TT"]:
|
83 |
tokens_ids = model.generate(**input_data, generate_speech=False, tgt_lang=target_language_code, num_beams=5, do_sample=True)[0].cpu().squeeze().detach().tolist()
|
84 |
else:
|
85 |
+
output = model.generate(**input_data, return_intermediate_token_ids=True, tgt_lang=target_language_code, num_beams=5, do_sample=True)
|
86 |
|
87 |
waveform = output.waveform.cpu().squeeze().detach().numpy()
|
88 |
tokens_ids = output.sequences.cpu().squeeze().detach().tolist()
|