leo-kwan commited on
Commit
b667341
1 Parent(s): 37ec437

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def translate(audio):
39
  tokenizer=tokenizer,
40
  device=device
41
  )
42
- outputs = asr_pipe(audio, max_new_tokens=256, generate_kwargs={"task": "translate", "forced_decoder_ids": forced_decoder_ids})
43
  return outputs["text"]
44
 
45
 
 
39
  tokenizer=tokenizer,
40
  device=device
41
  )
42
+ outputs = asr_pipe(audio, max_new_tokens=256, generate_kwargs={"task": "transcribe", "forced_decoder_ids": forced_decoder_ids})
43
  return outputs["text"]
44
 
45