Update handler.py

handler.py

@@ -4,6 +4,9 @@ from torchaudio.pipelines import SQUIM_OBJECTIVE
 import numpy as np
 from typing import Dict, Union, Any
 from io import BytesIO
+import logging
+
+logger = logging.getLogger(__name__)
 
 
 class EndpointHandler:
@@ -41,6 +44,7 @@ class EndpointHandler:
                 audio_buffer = BytesIO(input_data)
                 waveform, sample_rate = torchaudio.load(audio_buffer)
             elif isinstance(input_data, dict):
+                logger.info(f"Input data: {input_data}")
                 if 'audio' in input_data:
                     # Handle numpy array input
                     audio_array = input_data['audio']