Update handler.py

handler.py

@@ -31,6 +31,6 @@ class EndpointHandler:
             outputs = self.model.generate(**inputs)
 
         # postprocess the prediction
-        prediction = outputs[0].cpu().numpy()
+        prediction = outputs[0, 0].cpu().numpy()
 
         return [{"generated_audio": prediction}]