Update handler.py

handler.py

@@ -84,7 +84,7 @@ class EndpointHandler():
 
         result = {
             "final_image": image2,
-            "pipe1_img": out,
+            "pipe1_img": out.images[0],
             "pipe1_img_resized": out.images[0].resize((1024, 1024)),
             "pipe2_img": image
         }