Update handler.py

handler.py

@@ -14,7 +14,7 @@ class EndpointHandler():
         # self.pipe.model.config.forced_decoder_ids = self.pipe.model.processor.get_decoder_prompt_ids(language="Slovenian", task="transcribe")
         # self.pipe.model.generation_config.forced_decoder_ids = self.pipe.model.config.forced_decoder_ids  
 
-    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
+    def __call__(self, data: Dict[str, Any], **kwargs) -> List[Dict[str, Any]]:
         """
        data args:
             inputs (:obj: `str` | `PIL.Image` | `np.array`)
@@ -23,6 +23,7 @@ class EndpointHandler():
             A :obj:`list` | `dict`: will be serialized and returned
         """
         inputs = data.pop("inputs",data)
+        language = kwargs.get("language", "sl")
         # print("inputs", inputs)
-        prediction = self.pipe(inputs, generate_kwargs={"language": "Slovenian", "task": "transcribe"})
+        prediction = self.pipe(inputs, generate_kwargs={"language": language, "task": "transcribe"})
         return prediction