Update handler.py

handler.py

@@ -3,6 +3,7 @@ import json
 import numpy as np
 from transformers import AutoProcessor, MusicgenForConditionalGeneration
 import torch
+import logging
 
 
 class EndpointHandler:
@@ -23,17 +24,22 @@ class EndpointHandler:
             data (:dict:):
                 The payload with the text prompt and generation parameters.
         """
+        # Set up logging
+        logging.basicConfig(level=logging.DEBUG)
+        logger = logging.getLogger(__name__)
+        
         # process input
-        print(data)
-        data = json.loads(data)
+        logger.debug(f"Data: {data}")
         inputs = data.pop("inputs", data)
-        print(inputs)
+        logger.debug(f"Inputs: {inputs}")
         parameters = data.pop("parameters", None)
-        print(parameters)
+        logger.debug(f"Parameters: {parameters}")
         duration = parameters.pop("duration", None)
-        print(duration)
+        logger.debug(f"Duration: {duration}")
         audio = parameters.pop("audio", None)
+        logger.debug(f"Audio: {audio}")
         sampling_rate = parameters.pop("sampling_rate", None)
+        logger.debug(f"Sampling Rate: {sampling_rate}")
 
         if audio is not None:
             audio_list = json.loads(audio)