mariesig commited on
Commit
c4c5df8
·
1 Parent(s): df08ce3

reorder/name to match offline pipeline

Browse files
Files changed (1) hide show
  1. stream_pipeline.py +8 -8
stream_pipeline.py CHANGED
@@ -54,24 +54,24 @@ def shutdown_streamers():
54
  Streamer_raw = None
55
 
56
 
57
- def set_stt_streamer(model_name: str, fs_hz: int):
58
  global Streamer_enhanced, Streamer_raw
59
- StreamerCls = STREAMER_CLASSES.get(model_name, DeepgramStreamer)
60
  try:
61
  Streamer_enhanced = StreamerCls(
62
- fs_hz=fs_hz,
63
- stream_name="enhanced",
64
  on_update=_set_transcript_enhanced,
65
  )
66
  Streamer_raw = StreamerCls(
67
- fs_hz=fs_hz,
68
- stream_name="raw",
69
  on_update=_set_transcript_raw,
70
  )
71
  except Exception as e:
72
  Streamer_enhanced = None
73
  Streamer_raw = None
74
- raise RuntimeError(f"Error initializing STT streamer '{model_name}': {e}")
75
 
76
  def _to_float32_mono(y: np.ndarray) -> np.ndarray:
77
  y = np.asarray(y)
@@ -104,7 +104,7 @@ def _ensure_initialized(sr: int, streaming_sr, stt_model: str, enhancement_level
104
  enhancement_level=enhancement_level
105
  )
106
  SDK_STREAMING.init_processor(sdk_params)
107
- set_stt_streamer(stt_model, sr)
108
  return sr, *render_system_status("ready")
109
  except Exception as e:
110
  gr.Warning(f"Streaming process failed: {e}")
 
54
  Streamer_raw = None
55
 
56
 
57
+ def set_stt_streamer(sample_rate: int, stt_model: str):
58
  global Streamer_enhanced, Streamer_raw
59
+ StreamerCls = STREAMER_CLASSES.get(stt_model, DeepgramStreamer)
60
  try:
61
  Streamer_enhanced = StreamerCls(
62
+ fs_hz=sample_rate,
63
+ stream_name="Enhanced",
64
  on_update=_set_transcript_enhanced,
65
  )
66
  Streamer_raw = StreamerCls(
67
+ fs_hz=sample_rate,
68
+ stream_name="Raw",
69
  on_update=_set_transcript_raw,
70
  )
71
  except Exception as e:
72
  Streamer_enhanced = None
73
  Streamer_raw = None
74
+ raise RuntimeError(f"Error initializing STT streamer '{stt_model}': {e}")
75
 
76
  def _to_float32_mono(y: np.ndarray) -> np.ndarray:
77
  y = np.asarray(y)
 
104
  enhancement_level=enhancement_level
105
  )
106
  SDK_STREAMING.init_processor(sdk_params)
107
+ set_stt_streamer(sr, stt_model)
108
  return sr, *render_system_status("ready")
109
  except Exception as e:
110
  gr.Warning(f"Streaming process failed: {e}")