Spaces:
Sleeping
Sleeping
Fix syntax error in TTS stage and complete pipeline
Browse files
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
|
| 411 |
-
|
| 412 |
-
|
| 413 |
-
|
| 414 |
-
|
|
|
|
| 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()
|