Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 =
|
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 |
|