MoHamdyy commited on
Commit
aa4e827
·
1 Parent(s): 43571f5

Fix syntax error in TTS stage and complete pipeline

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -407,11 +407,12 @@ def full_speech_translation_pipeline(audio_input_path: str):
407
  forced_ids = stt_processor.get_decoder_prompt_ids(language="arabic", task="transcribe")
408
 
409
  with torch.no_grad():
410
- # Pass forced_decoder_ids via a GenerationConfig to avoid unused kwargs error
411
- generation_config = stt_model.generation_config.copy()
412
- generation_config.forced_decoder_ids = forced_ids
413
-
414
- generated_ids = stt_model.generate(inputs, generation_config=generation_config, max_length=448)
 
415
 
416
  # Use batch_decode for robustness
417
  arabic_transcript = stt_processor.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()
 
407
  forced_ids = stt_processor.get_decoder_prompt_ids(language="arabic", task="transcribe")
408
 
409
  with torch.no_grad():
410
+ # Pass forced_decoder_ids directly as a keyword argument
411
+ generated_ids = stt_model.generate(
412
+ inputs,
413
+ forced_decoder_ids=forced_ids,
414
+ max_length=448
415
+ )
416
 
417
  # Use batch_decode for robustness
418
  arabic_transcript = stt_processor.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()