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