Update handler.py

handler.py

@@ -162,15 +162,21 @@ class EndpointHandler():
         else:
             return None
 
-    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
-        """
-        The actual method called during inference. Expects data to have a 'path' to the audio file.
-        """
-        # Get the path to the audio file from the request data
-        path = data.get("path")
-
-        # If the path is provided, we run the prediction, else return an error message
-        if path:
-            return self.predict(path)
+    def __call__(self, request: Dict[str, Any]) -> List[Dict[str, Any]]:
+        # Get the binary content of the audio file (assuming it is passed as 'inputs')
+        audio_data = request.get("inputs")
+
+        if audio_data:
+            # Convert binary content to a bytes buffer
+            audio_buffer = io.BytesIO(audio_data)
+            
+            # However, since this handler is not loading from a path (as it does in speech_file_to_array_fn), 
+            # we need to read and process the buffer similar to how speech_file_to_array_fn would
+            waveform, sample_rate = torchaudio.load(audio_buffer)
+            waveform = waveform.squeeze().numpy()
+
+            # Call the predict function and return its results
+            predictions = self.predict(waveform, sample_rate)
+            return predictions
         else:
-            return {"error": "Path to the audio file is required."}
+            return {"error": "Audio input is required."}