Adriaanvh1 commited on
Commit
a295a8d
·
1 Parent(s): 76fc7e6

required inputs, base64 encoding

Browse files
Files changed (1) hide show
  1. handler.py +4 -3
handler.py CHANGED
@@ -31,9 +31,10 @@ class EndpointHandler:
31
  Returns:
32
  List[Dict[str, Any]]: _description_
33
  """
34
- audio = data.pop("audio") # bytes
35
- lang = data.pop("language")
36
- task = data.pop("task") # One of "translate", "transcribe"
 
37
 
38
  # Set language and task
39
  self.pipeline.model.config.forced_decoder_ids = self.pipeline.tokenizer.get_decoder_prompt_ids(language=lang, task=task)
 
31
  Returns:
32
  List[Dict[str, Any]]: _description_
33
  """
34
+ inputs = data["inputs"]
35
+ audio = inputs["audio"] # bytes
36
+ lang = inputs["language"]
37
+ task = inputs["task"] # One of "translate", "transcribe"
38
 
39
  # Set language and task
40
  self.pipeline.model.config.forced_decoder_ids = self.pipeline.tokenizer.get_decoder_prompt_ids(language=lang, task=task)