lingdoc commited on
Commit
744dbe2
·
verified ·
1 Parent(s): cb4b944

Update handler.py

Browse files
Files changed (1) hide show
  1. 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, providing defaults if necessary
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