Update handler.py

handler.py

@@ -1,4 +1,6 @@
 from typing import Dict, List, Any
+import json
+import numpy as np
 from transformers import AutoProcessor, MusicgenForConditionalGeneration
 import torch
 
@@ -25,6 +27,12 @@ class EndpointHandler:
         inputs = data.pop("inputs", data)
         parameters = data.pop("parameters", None)
         duration = parameters.pop("duration", None)
+        audio = parameters.pop("audio", None)
+
+        if audio is not None:
+            audio_list = json.loads(audio)
+            audio_array = np.array(audio_list)
+            parameters["audio"] = audio_array
 
         if duration is not None:
             # Calculate max new tokens based on duration, this is a placeholder, replace with actual logic