Update app.py

app.py

@@ -158,7 +158,7 @@ def execute(image, prompt, debug=False):
     im =  Image.fromarray(rgb_image)
     person = Image.open('mask_person.jpg')
 
-    result = pipe(
+    result0 = pipe(
             prompt=f"{prompt} {TRIGGER}",
             image=im,
             mask_image=person,
@@ -171,7 +171,7 @@ def execute(image, prompt, debug=False):
             joint_attention_kwargs={"scale": 0.9},
         ).images[0]
 
-    arr = np.array(result)
+    arr = np.array(result0)
     rgb_image = cv2.cvtColor(arr, cv2.COLOR_BGR2RGB)
     cv2.imwrite('person.jpg', rgb_image)
     cv2.imwrite("mask.jpg", maskHead('person.jpg'))
@@ -179,7 +179,7 @@ def execute(image, prompt, debug=False):
 
     result = pipe(
             prompt=f"{prompt} {TRIGGER}",
-            image=result,
+            image=result0,
             mask_image=mask,
             width=1024,
             height=1024,
@@ -194,7 +194,9 @@ def execute(image, prompt, debug=False):
         response.append(im)
         response.append(person)
         response.append(mask)
+        response.append(result0)
 
+    response.append(result)
   return response
 
 # Created by Fountai