Update handler.py

handler.py

@@ -30,13 +30,14 @@ class EndpointHandler():
         buffer = io.BytesIO()
         sf.write(buffer, full_audio.numpy(), 24000, format='WAV')
         buffer.seek(0)
-        
-        from flask import Response
-        return Response(
-            buffer.read(),
-            mimetype='audio/wav',
-            headers={
-                'Content-Disposition': 'attachment; filename=output.wav',
-                'Content-Type': 'audio/wav'
-            }
-        )
+        audio_bytes = buffer.read()
+        audio_b64 = base64.b64encode(audio_bytes).decode("utf-8")
+        return {
+            "headers": {
+                "Content-Disposition": "attachment; filename=output.wav",
+                "Content-Type": "audio/wav"
+            },
+            "body": audio_bytes,
+            "statusCode": 200,
+            "isBase64Encoded": True  # if binary, encode
+        }