leofltt commited on
Commit
3999b52
·
verified ·
1 Parent(s): ae00001

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def translate(audio):
23
  inputs = asr_processor(audio, sampling_rate=SAMPLE_RATE, return_tensors="pt")
24
  generated_ids = asr_model.generate(inputs["input_features"],attention_mask=inputs["attention_mask"],
25
  forced_bos_token_id=asr_processor.tokenizer.lang_code_to_id["it"],)
26
- translation = processor.batch_decode(generated_ids, skip_special_tokens=True)
27
  return translation
28
 
29
 
 
23
  inputs = asr_processor(audio, sampling_rate=SAMPLE_RATE, return_tensors="pt")
24
  generated_ids = asr_model.generate(inputs["input_features"],attention_mask=inputs["attention_mask"],
25
  forced_bos_token_id=asr_processor.tokenizer.lang_code_to_id["it"],)
26
+ translation = asr_processor.batch_decode(generated_ids, skip_special_tokens=True)
27
  return translation
28
 
29