jj-wohlgemuth commited on
Commit
8b29559
·
1 Parent(s): 4f24eb5

renamed STT names

Browse files
Files changed (2) hide show
  1. app.py +12 -6
  2. constants.py +4 -4
app.py CHANGED
@@ -35,11 +35,11 @@ with gr.Blocks() as demo:
35
  # ✅ Global controls (shared by both tabs)
36
  stt_model = gr.Radio(
37
  label="STT Model",
38
- choices=["Soniox", "Deepgram"],
39
- value="Soniox",
40
  interactive=True,
41
  )
42
- set_stt_streamer(stt_model.value)
43
  enhancement_level = gr.Slider(
44
  minimum=0,
45
  maximum=100,
@@ -69,18 +69,24 @@ with gr.Blocks() as demo:
69
  with gr.Row():
70
  gr.Markdown(open("docs/dawn_chorus.md", "r", encoding="utf-8").read())
71
  dataset_dropdown = gr.Dropdown(choices=ALL_FILES, value=None)
72
- audio_file_from_dataset = gr.Audio(type="filepath", interactive=False, show_download_button=True)
 
 
73
 
74
  with gr.Tab("Upload local file") as upload_tab:
75
  with gr.Row():
76
  gr.Markdown(open("docs/local_file.md", "r", encoding="utf-8").read())
77
- audio_file_upload = gr.Audio(type="filepath", sources=["upload"], show_download_button=True)
 
 
78
 
79
  enhance_btn = gr.Button("Enhance", scale=2, visible=False)
80
 
81
  with gr.Group(elem_classes="panel results-card", visible=False) as results_card:
82
  result_title = gr.Markdown("")
83
- enhanced_audio = gr.Audio(type="numpy", interactive=False, show_download_button=True)
 
 
84
 
85
  with gr.Row(equal_height=True, elem_classes="results-row"):
86
  with gr.Column(scale=5, min_width=320, elem_classes="results-left"):
 
35
  # ✅ Global controls (shared by both tabs)
36
  stt_model = gr.Radio(
37
  label="STT Model",
38
+ choices=["Soniox STT-RT v3", "Deepgram Nova-3 RT"],
39
+ value="Soniox STT-RT v3",
40
  interactive=True,
41
  )
42
+ set_stt_streamer("Soniox STT-RT v3")
43
  enhancement_level = gr.Slider(
44
  minimum=0,
45
  maximum=100,
 
69
  with gr.Row():
70
  gr.Markdown(open("docs/dawn_chorus.md", "r", encoding="utf-8").read())
71
  dataset_dropdown = gr.Dropdown(choices=ALL_FILES, value=None)
72
+ audio_file_from_dataset = gr.Audio(
73
+ type="filepath", interactive=False, show_download_button=True
74
+ )
75
 
76
  with gr.Tab("Upload local file") as upload_tab:
77
  with gr.Row():
78
  gr.Markdown(open("docs/local_file.md", "r", encoding="utf-8").read())
79
+ audio_file_upload = gr.Audio(
80
+ type="filepath", sources=["upload"], show_download_button=True
81
+ )
82
 
83
  enhance_btn = gr.Button("Enhance", scale=2, visible=False)
84
 
85
  with gr.Group(elem_classes="panel results-card", visible=False) as results_card:
86
  result_title = gr.Markdown("")
87
+ enhanced_audio = gr.Audio(
88
+ type="numpy", interactive=False, show_download_button=True
89
+ )
90
 
91
  with gr.Row(equal_height=True, elem_classes="results-row"):
92
  with gr.Column(scale=5, min_width=320, elem_classes="results-left"):
constants.py CHANGED
@@ -10,7 +10,7 @@ MODEL_ID: Final = "quail-vf-2.0-l-16khz"
10
 
11
  MINUTES_KEEP: Final = 60
12
 
13
- DATASET_NAME: Final = "ai-coustics/dawn_chorus_en"
14
  DEFAULT_SPLIT: Final = "eval"
15
  MIX_DIR: Final = "mix"
16
  SPEECH_DIR: Final = "speech"
@@ -20,6 +20,6 @@ DEFAULT_SR: Final = 16000
20
  STREAM_EVERY: Final = 0.2
21
 
22
  STREAMER_CLASSES: Final = {
23
- "Deepgram": DeepgramStreamer,
24
- "Soniox": SonioxStreamer,
25
- }
 
10
 
11
  MINUTES_KEEP: Final = 60
12
 
13
+ DATASET_NAME: Final = "ai-coustics/dawn_chorus_en"
14
  DEFAULT_SPLIT: Final = "eval"
15
  MIX_DIR: Final = "mix"
16
  SPEECH_DIR: Final = "speech"
 
20
  STREAM_EVERY: Final = 0.2
21
 
22
  STREAMER_CLASSES: Final = {
23
+ "Deepgram Nova-3 RT": DeepgramStreamer,
24
+ "Soniox STT-RT v3": SonioxStreamer,
25
+ }