correction

call.py

@@ -0,0 +1,29 @@
+import requests
+import base64
+import json
+
+API_URL = "https://w4a3d6hyuuiyrznb.eu-west-1.aws.endpoints.huggingface.cloud"
+headers = {
+	"Authorization": "Bearer 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",
+	"Content-Type": "application/json"
+}
+
+def query(filename):
+    with open(filename, "rb") as f:
+        audio_data = f.read()
+    print(audio_data)
+    # Encode the audio data as base64
+    audio_base64 = base64.b64encode(audio_data).decode("utf-8")
+    # print(audio_base64)
+    
+    # Construct the JSON payload
+    payload = {
+        "inputs": audio_data,
+    }
+    
+    response = requests.post(API_URL, headers=headers, data=payload)
+    return response.json()
+
+output = query("sample1.flac")
+print("-----------------")
+print(output)

handler.py

@@ -126,7 +126,7 @@ class EndpointHandler():
         logger.info(f"Model for diarization initialized")
 
         
-    def __call__(self, data: Dict[str, bytes]) -> Dict[str, str]:
+    def __call__(self, data: Any) -> Dict[str, str]:
             """
             Args:
                 data (:obj:):
@@ -141,7 +141,16 @@ class EndpointHandler():
                 print(f"key: {x}, value: {data[x]}    ")
             
             # 1. process input
-            inputs_encoded = data.pop("inputs", data)
+            inputs = data.pop("inputs", data)
+            parameters = data.pop("parameters", None)
+        
+            inputs_encoded = inputs["audio_encoded"]
+            print(inputs_encoded)
+
+            language = None
+            if "language" in inputs.keys():
+                language = inputs["language"]
+
             inputs = base64.b64decode(inputs_encoded)
             
             print(inputs)