Spaces:

kadirnar
/

Whisper_M2M100_BioGpt

Sleeping

kadirnar commited on Feb 20, 2023

Commit

b564f38

•

1 Parent(s): 1b52738

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ def whisper_demo(input_audio, model_id):
     pipe = pipeline(task="automatic-speech-recognition",model=model_id, device='cuda:0')
     pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language='en', task="transcribe")
     output_text = pipe(input_audio)['text']
-    return output_text[0]['generated_text']
 def translate_to_english(prompt, lang_model_id, base_lang):

     pipe = pipeline(task="automatic-speech-recognition",model=model_id, device='cuda:0')
     pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language='en', task="transcribe")
     output_text = pipe(input_audio)['text']
+    return output_text[0]
 def translate_to_english(prompt, lang_model_id, base_lang):