lihaoxin2020 commited on
Commit
365b3e7
·
verified ·
1 Parent(s): e7e2937

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, spkr_id=LANG_TO_SPKR_ID[target_language_code][0])
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()