Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -84,16 +84,15 @@ def transcribe(audio, task_type, language):
|
|
84 |
|
85 |
# Use the pipeline to perform transcription
|
86 |
sample = {"array": waveform.squeeze().numpy(), "sampling_rate": sample_rate}
|
87 |
-
|
88 |
"task": task_type,
|
89 |
"num_beams": 3,
|
90 |
-
"num_return_sequences": 3,
|
91 |
"no_repeat_ngram_size": 4,
|
92 |
"early_stopping": True,
|
93 |
"language": language
|
94 |
-
})
|
95 |
-
|
96 |
-
return
|
97 |
|
98 |
def get_wav_files(directory):
|
99 |
"""
|
|
|
84 |
|
85 |
# Use the pipeline to perform transcription
|
86 |
sample = {"array": waveform.squeeze().numpy(), "sampling_rate": sample_rate}
|
87 |
+
text = pipe(sample, generate_kwargs={
|
88 |
"task": task_type,
|
89 |
"num_beams": 3,
|
|
|
90 |
"no_repeat_ngram_size": 4,
|
91 |
"early_stopping": True,
|
92 |
"language": language
|
93 |
+
})["text"]
|
94 |
+
|
95 |
+
return text
|
96 |
|
97 |
def get_wav_files(directory):
|
98 |
"""
|