logs

handler.py

@@ -1,3 +1,4 @@
+import logging
 from speechbrain.pretrained import EncoderClassifier
 from typing import Dict, List, Any
 
@@ -6,6 +7,7 @@ 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]) -> List[Dict[str, Any]]:
         """
@@ -19,6 +21,7 @@ class EndpointHandler:
         audio_url = data.pop("audio_url", data)
 
         print('audio_url', audio_url)
+        logging.info(f'audio_url {audio_url}')
 
         # check if date exists and if it is a holiday
         # run normal prediction