leofltt commited on
Commit
c909167
·
verified ·
1 Parent(s): 7219472

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ bark_model.to(device)
24
 
25
  def translate(audio):
26
  inputs = asr_processor(audio, sampling_rate=16000, return_tensors="pt")
27
- output_tokens = asr_model.generate(**inputs, tgt_lang="it", generate_speech=False)
28
  translation = asr_processor.decode(output_tokens[0].tolist()[0], skip_special_tokens=True)
29
  return translation
30
 
 
24
 
25
  def translate(audio):
26
  inputs = asr_processor(audio, sampling_rate=16000, return_tensors="pt")
27
+ output_tokens = asr_model.generate(**inputs, tgt_lang="ita", generate_speech=False)
28
  translation = asr_processor.decode(output_tokens[0].tolist()[0], skip_special_tokens=True)
29
  return translation
30