Update handler.py

handler.py

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