Banafo commited on
Commit
dcdecd0
·
verified ·
1 Parent(s): 4713af7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -1
app.py CHANGED
@@ -26,6 +26,16 @@ recognizer_fr = OnlineRecognizer.from_transducer(
26
  debug=False
27
  )
28
 
 
 
 
 
 
 
 
 
 
 
29
  def transcribe_audio_online_streaming(file, language):
30
  """Generator for file transcription"""
31
  if file is None:
@@ -38,6 +48,8 @@ def transcribe_audio_online_streaming(file, language):
38
  recognizer = recognizer_en
39
  case "French":
40
  recognizer = recognizer_fr
 
 
41
 
42
  waveform, sample_rate = torchaudio.load(file.name)
43
  if sample_rate != 16000:
@@ -84,6 +96,8 @@ def transcribe_microphone_stream(audio_chunk, stream_state, language):
84
  recognizer = recognizer_en
85
  case "French":
86
  recognizer = recognizer_fr
 
 
87
 
88
  if audio_chunk is None: # End of stream
89
  if stream_state is not None:
@@ -129,7 +143,7 @@ def transcribe_microphone_stream(audio_chunk, stream_state, language):
129
  def create_app():
130
  with gr.Blocks() as app:
131
  gr.Markdown("# Real-time Speech Recognition")
132
- language_choice = gr.Radio(choices=["English", "French"], label="Select Language", value="English")
133
 
134
  with gr.Tabs():
135
  with gr.Tab("File Transcription"):
 
26
  debug=False
27
  )
28
 
29
+ recognizer_de = OnlineRecognizer.from_transducer(
30
+ tokens="de_tokens.txt",
31
+ encoder="de_encoder.onnx",
32
+ decoder="de_decoder.onnx",
33
+ joiner="de_joiner.onnx",
34
+ num_threads=1,
35
+ decoding_method="modified_beam_search",
36
+ debug=False
37
+ )
38
+
39
  def transcribe_audio_online_streaming(file, language):
40
  """Generator for file transcription"""
41
  if file is None:
 
48
  recognizer = recognizer_en
49
  case "French":
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:
 
96
  recognizer = recognizer_en
97
  case "French":
98
  recognizer = recognizer_fr
99
+ case "German":
100
+ recognizer = recognizer_de
101
 
102
  if audio_chunk is None: # End of stream
103
  if stream_state is not None:
 
143
  def create_app():
144
  with gr.Blocks() as app:
145
  gr.Markdown("# Real-time Speech Recognition")
146
+ language_choice = gr.Radio(choices=["English", "French", "German"], label="Select Language", value="English")
147
 
148
  with gr.Tabs():
149
  with gr.Tab("File Transcription"):