Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -84,9 +84,16 @@ 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 |
-
|
89 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
90 |
|
91 |
def get_wav_files(directory):
|
92 |
"""
|
|
|
84 |
|
85 |
# Use the pipeline to perform transcription
|
86 |
sample = {"array": waveform.squeeze().numpy(), "sampling_rate": sample_rate}
|
87 |
+
texts = pipe(sample, generate_kwargs={
|
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 |
+
print(texts)
|
96 |
+
return texts[0]['text']
|
97 |
|
98 |
def get_wav_files(directory):
|
99 |
"""
|