speech-to-speech-translation

Running

Marco-Cheung commited on Aug 19, 2023

Commit

e9f9b4b

•

1 Parent(s): 60ec677

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ model = BarkModel.from_pretrained("suno/bark-small").to(device)
 sampling_rate = model.generation_config.sample_rate
 # set the forced ids
-model.config.forced_decoder_ids = processor.get_decoder_prompt_ids(task="translate")
 def translate(audio):
     outputs = asr_pipe(audio, max_new_tokens=256, generate_kwargs={"task": "translate"})

 sampling_rate = model.generation_config.sample_rate
 # set the forced ids
+model.generation_config.forced_decoder_ids = processor.tokenizer.get_decoder_prompt_ids(language='de', task="translate")
 def translate(audio):
     outputs = asr_pipe(audio, max_new_tokens=256, generate_kwargs={"task": "translate"})