Update app.py
Browse filesAdd Spanish to options
app.py
CHANGED
@@ -36,6 +36,16 @@ recognizer_de = OnlineRecognizer.from_transducer(
|
|
36 |
debug=False
|
37 |
)
|
38 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
39 |
def transcribe_audio_online_streaming(file, language):
|
40 |
"""Generator for file transcription"""
|
41 |
if file is None:
|
@@ -50,6 +60,8 @@ def transcribe_audio_online_streaming(file, language):
|
|
50 |
recognizer = recognizer_fr
|
51 |
case "German":
|
52 |
recognizer = recognizer_de
|
|
|
|
|
53 |
|
54 |
waveform, sample_rate = torchaudio.load(file.name)
|
55 |
if sample_rate != 16000:
|
@@ -112,6 +124,8 @@ def transcribe_microphone_stream(audio_chunk, stream_state, language):
|
|
112 |
recognizer = recognizer_fr
|
113 |
case "German":
|
114 |
recognizer = recognizer_de
|
|
|
|
|
115 |
|
116 |
if audio_chunk is None: # End of stream
|
117 |
if stream_state is not None:
|
@@ -164,7 +178,7 @@ def transcribe_microphone_stream(audio_chunk, stream_state, language):
|
|
164 |
def create_app():
|
165 |
with gr.Blocks() as app:
|
166 |
gr.Markdown("# Real-time Speech Recognition")
|
167 |
-
language_choice = gr.Radio(choices=["English", "French", "German"], label="Select Language", value="English")
|
168 |
|
169 |
with gr.Tabs():
|
170 |
with gr.Tab("File Transcription"):
|
|
|
36 |
debug=False
|
37 |
)
|
38 |
|
39 |
+
recognizer_es = OnlineRecognizer.from_transducer(
|
40 |
+
tokens="es_tokens.txt",
|
41 |
+
encoder="es_encoder.onnx",
|
42 |
+
decoder="es_decoder.onnx",
|
43 |
+
joiner="es_joiner.onnx",
|
44 |
+
num_threads=1,
|
45 |
+
decoding_method="modified_beam_search",
|
46 |
+
debug=False
|
47 |
+
)
|
48 |
+
|
49 |
def transcribe_audio_online_streaming(file, language):
|
50 |
"""Generator for file transcription"""
|
51 |
if file is None:
|
|
|
60 |
recognizer = recognizer_fr
|
61 |
case "German":
|
62 |
recognizer = recognizer_de
|
63 |
+
case "Spanish":
|
64 |
+
recognizer = recognizer_es
|
65 |
|
66 |
waveform, sample_rate = torchaudio.load(file.name)
|
67 |
if sample_rate != 16000:
|
|
|
124 |
recognizer = recognizer_fr
|
125 |
case "German":
|
126 |
recognizer = recognizer_de
|
127 |
+
case "Spanish":
|
128 |
+
recognizer = recognizer_es
|
129 |
|
130 |
if audio_chunk is None: # End of stream
|
131 |
if stream_state is not None:
|
|
|
178 |
def create_app():
|
179 |
with gr.Blocks() as app:
|
180 |
gr.Markdown("# Real-time Speech Recognition")
|
181 |
+
language_choice = gr.Radio(choices=["English", "French", "German", "Spanish"], label="Select Language", value="English")
|
182 |
|
183 |
with gr.Tabs():
|
184 |
with gr.Tab("File Transcription"):
|