Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
mariesig commited on
Commit ·
b80bb23
1
Parent(s): abfb82b
Remove unused function
Browse files
utils.py
CHANGED
|
@@ -93,43 +93,4 @@ def compute_wer(reference: str, hypothesis: str) -> float:
|
|
| 93 |
d[i - 1][j - 1] + cost, # Substitution
|
| 94 |
)
|
| 95 |
wer = d[len(ref_words)][len(hyp_words)] / max(len(ref_words), 1)
|
| 96 |
-
return wer
|
| 97 |
-
|
| 98 |
-
|
| 99 |
-
|
| 100 |
-
def transcribe_audio(
|
| 101 |
-
audio_array: np.ndarray,
|
| 102 |
-
sr: int,
|
| 103 |
-
streamer_type: str = "Deepgram",
|
| 104 |
-
stream_name: str = "RAW",
|
| 105 |
-
on_update: Optional[Callable[[str], None]] = None,
|
| 106 |
-
):
|
| 107 |
-
"""
|
| 108 |
-
Transcribe an audio array using the specified STT streamer.
|
| 109 |
-
|
| 110 |
-
Args:
|
| 111 |
-
audio_array (np.ndarray): Audio data array
|
| 112 |
-
sr (int): Sample rate of the audio array
|
| 113 |
-
streamer_type (str): "Soniox" or "Deepgram"
|
| 114 |
-
stream_name (str): Optional label for streamer instance ("RAW", "ENHANCED", etc.)
|
| 115 |
-
on_update: Optional callback(text: str) called with partial transcript as results stream in.
|
| 116 |
-
|
| 117 |
-
Returns:
|
| 118 |
-
str: Final transcript text
|
| 119 |
-
"""
|
| 120 |
-
if sr != DEFAULT_SR:
|
| 121 |
-
audio_array = resampy.resample(audio_array, sr, DEFAULT_SR)
|
| 122 |
-
sr = DEFAULT_SR
|
| 123 |
-
|
| 124 |
-
if streamer_type not in STREAMER_CLASSES:
|
| 125 |
-
raise ValueError(
|
| 126 |
-
f"Invalid streamer_type '{streamer_type}'. "
|
| 127 |
-
f"Choose from: {', '.join(STREAMER_CLASSES.keys())}"
|
| 128 |
-
)
|
| 129 |
-
|
| 130 |
-
StreamerClass = STREAMER_CLASSES[streamer_type]
|
| 131 |
-
streamer = StreamerClass(sr, stream_name, on_update=on_update)
|
| 132 |
-
|
| 133 |
-
transcript = streamer.stream_array(audio_array)
|
| 134 |
-
|
| 135 |
-
return transcript
|
|
|
|
| 93 |
d[i - 1][j - 1] + cost, # Substitution
|
| 94 |
)
|
| 95 |
wer = d[len(ref_words)][len(hyp_words)] / max(len(ref_words), 1)
|
| 96 |
+
return wer
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|