Update speaker_id

app.py

@@ -82,7 +82,7 @@ def predict(
     if task_name in ["S2TT", "T2TT"]:
         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()
     else:
-        output = model.generate(**input_data, return_intermediate_token_ids=True, tgt_lang=target_language_code, num_beams=5, do_sample=True)
+        output = model.generate(**input_data, return_intermediate_token_ids=True, tgt_lang=target_language_code, speaker_id=LANGUAGE_NAME_TO_CODE[target_language_code], num_beams=5, do_sample=True)
         
         waveform = output.waveform.cpu().squeeze().detach().numpy()
         tokens_ids = output.sequences.cpu().squeeze().detach().tolist()