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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ bark_processor = BarkProcessor.from_pretrained("suno/bark")
20
 
21
 
22
  def translate(audio):
23
- inputs = 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)
 
20
 
21
 
22
  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)