Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
mariesig commited on
Commit ·
c4c5df8
1
Parent(s): df08ce3
reorder/name to match offline pipeline
Browse files- 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(
|
| 58 |
global Streamer_enhanced, Streamer_raw
|
| 59 |
-
StreamerCls = STREAMER_CLASSES.get(
|
| 60 |
try:
|
| 61 |
Streamer_enhanced = StreamerCls(
|
| 62 |
-
fs_hz=
|
| 63 |
-
stream_name="
|
| 64 |
on_update=_set_transcript_enhanced,
|
| 65 |
)
|
| 66 |
Streamer_raw = StreamerCls(
|
| 67 |
-
fs_hz=
|
| 68 |
-
stream_name="
|
| 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 '{
|
| 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(
|
| 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}")
|