Commit
·
a295a8d
1
Parent(s):
76fc7e6
required inputs, base64 encoding
Browse files- handler.py +4 -3
handler.py
CHANGED
|
@@ -31,9 +31,10 @@ class EndpointHandler:
|
|
| 31 |
Returns:
|
| 32 |
List[Dict[str, Any]]: _description_
|
| 33 |
"""
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
|
|
|
| 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)
|