updates handler.py

handler.py

@@ -6,17 +6,17 @@ from typing import Dict, List, Any
 class EndpointHandler:
     def __init__(self, path=""):
         self.model = EncoderClassifier.from_hparams("speechbrain/lang-id-voxlingua107-ecapa")
-        print('model loaded')
         logging.info('model loaded')
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
-        inputs = data.pop("inputs",data)
+        audio_url = data.pop("audio_url",data)
 
-        print('audio_url', inputs)
-        logging.info(f'audio_url {inputs}')
+        logging.info(f'audio_url {audio_url}')
 
-        # check if date exists and if it is a holiday
-        # run normal prediction
-        output = self.model.classify_file(inputs)
-        return {"prediction": output[1].exp()}
+        output = self.model.classify_file(audio_url)
+        
+        return {
+            "prediction": float(output[1].exp()[0]),
+            "language": output[3][0],
+        }