Spaces:
Sleeping
Sleeping
fix batch decoding
Browse files
app.py
CHANGED
@@ -34,9 +34,9 @@ def predict_and_ctc_lm_decode(input_file, model_name):
|
|
34 |
input_values = processor(speech, return_tensors="pt", sampling_rate=16000).input_values
|
35 |
logits = model(input_values).logits.cpu().detach().numpy()[0]
|
36 |
|
37 |
-
pred = processor.
|
38 |
|
39 |
-
transcribed_text = fix_transcription_casing(pred
|
40 |
|
41 |
return transcribed_text
|
42 |
|
|
|
34 |
input_values = processor(speech, return_tensors="pt", sampling_rate=16000).input_values
|
35 |
logits = model(input_values).logits.cpu().detach().numpy()[0]
|
36 |
|
37 |
+
pred = processor.decode(logits).text
|
38 |
|
39 |
+
transcribed_text = fix_transcription_casing(pred.lower())
|
40 |
|
41 |
return transcribed_text
|
42 |
|