Update handler.py
Browse files- handler.py +2 -10
handler.py
CHANGED
|
@@ -21,18 +21,10 @@ class EndpointHandler:
|
|
| 21 |
# process input
|
| 22 |
inputs = data.pop("inputs", data)
|
| 23 |
audio_nparray = ffmpeg_read(inputs, 16000)
|
| 24 |
-
# Retrieve custom arguments
|
| 25 |
params = data.pop("parameters", {})
|
| 26 |
-
language = params.get("language", "en")
|
| 27 |
-
vad_filter = params.get("vad_filter", True)
|
| 28 |
-
vad_params = params.get("vad_params", None)
|
| 29 |
-
batched = params.get("batched", True)
|
| 30 |
if batched:
|
| 31 |
-
segments, info = self.batched_model.transcribe(audio_nparray,
|
| 32 |
-
language=language, # can use this to constrain language, otherwise the language is detected from first 30 seconds
|
| 33 |
-
vad_filter=vad_filter,
|
| 34 |
-
vad_parameters=vad_params
|
| 35 |
-
)
|
| 36 |
else:
|
| 37 |
segments, info = self.model.transcribe(audio_nparray, beam_size=5)
|
| 38 |
|
|
|
|
| 21 |
# process input
|
| 22 |
inputs = data.pop("inputs", data)
|
| 23 |
audio_nparray = ffmpeg_read(inputs, 16000)
|
| 24 |
+
# Retrieve custom arguments
|
| 25 |
params = data.pop("parameters", {})
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
if batched:
|
| 27 |
+
segments, info = self.batched_model.transcribe(audio_nparray, **params)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
else:
|
| 29 |
segments, info = self.model.transcribe(audio_nparray, beam_size=5)
|
| 30 |
|