Update handler.py

handler.py

@@ -74,16 +74,23 @@ class EndpointHandler():
         
         self.pipe3.enable_xformers_memory_efficient_attention()
         
-        image = self.pipe3(
+        image2 = self.pipe3(
             prompt=prompt,
             image=image,
             guidance_scale=8.0,
             num_inference_steps=100,
             strength=0.2,
         ).images[0]
+
+        result = {
+            "final_image": image2,
+            "pipe1_img": out,
+            "pipe1_img_resized": out.images[0].resize((1024, 1024)),
+            "pipe2_img": image
+        }
             
         # return first generate PIL image
-        return image
+        return result
     
     # helper to decode input image
     def decode_base64_image(self, image_string):