numpy

handler.py

@@ -54,6 +54,12 @@ class EndpointHandler():
         if not query:
             return [{"error": "No query provided"}]
         
+        # Convert list to numpy array if needed (when sent via JSON)
+        if isinstance(audio, list):
+            audio = np.array(audio, dtype=np.float32)
+        elif not isinstance(audio, np.ndarray):
+            return [{"error": f"Audio data must be a list or numpy array, got {type(audio)}"}]
+        
         try:
             # Run inference using the pipeline
             results = self.pipeline(

requirements.txt

@@ -1,2 +1,5 @@
 git+https://github.com/earthspecies/beans-zero.git
-git+https://github.com/earthspecies/naturelm-audio.git@auth_tok
+git+https://github.com/earthspecies/naturelm-audio.git@auth_tok
+requests>=2.25.0
+soundfile>=0.10.0
+numpy>=1.21.0