Banafo commited on
Commit
9a22924
·
verified ·
1 Parent(s): 5756e90

Update app.py

Browse files

Add Spanish to options

Files changed (1) hide show
  1. app.py +15 -1
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"):