Update handler.py

handler.py

@@ -2,6 +2,10 @@ from typing import  Dict
 from pyannote.audio import Pipeline
 from transformers.pipelines.audio_utils import ffmpeg_read
 import torch 
+from transformers.utils import logging
+
+logger = logging.get_logger("MAXWELL")
+
 
 SAMPLE_RATE = 16000
 
@@ -21,8 +25,8 @@ class EndpointHandler():
             A :obj:`dict`:. base64 encoded image
         """
         # process input
-        print("MAXWELL")
-        print(data)
+        logger.warning("MAXWELL")
+        logger.warning(data)
         inputs = data.pop("inputs", data)
         parameters = data.pop("parameters", None) #  min_speakers=2, max_speakers=5