Update handler.py

handler.py

@@ -1,6 +1,7 @@
 import logging
 import torch
 import os
+import base64
 
 from pyannote.audio import Pipeline
 from transformers import pipeline, AutoModelForCausalLM
@@ -50,7 +51,10 @@ class EndpointHandler():
             self.diarization_pipeline = None
             
     
-    async def __call__(self, file, parameters):
+    async def __call__(self, inputs):
+        file = inputs.pop("inputs")
+        file = base64.b64decode(file)
+        parameters = inputs.pop("parameters", {})
         try:
             parameters = InferenceConfig(**parameters)
         except ValidationError as e: