Update handler.py

handler.py

@@ -35,4 +35,4 @@ class EndpointHandler:
         # postprocess the prediction
         prediction = outputs[0].cpu().numpy().tolist()
 
-        return [{"generated_audio": prediction}]
+        return [{"audio": prediction}]