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