inputs

handler.py

@@ -10,13 +10,13 @@ class EndpointHandler:
         logging.info('model loaded')
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
-        audio_url = data.pop("audio_url", data)
+        inputs = data.pop("inputs",data)
 
-        print('audio_url', audio_url)
-        logging.info(f'audio_url {audio_url}')
+        print('audio_url', inputs)
+        logging.info(f'audio_url {inputs}')
 
         # check if date exists and if it is a holiday
         # run normal prediction
-        output = self.model.classify_file(audio_url)
+        output = self.model.classify_file(inputs)
         return {"prediction": output[1].exp()}