.

app.py

@@ -29,7 +29,7 @@ def process_image(input_image, mode, weight1=None, weight2=None):
     elif mode == "refine":
         sotai_image, sketch_image = process_image_as_base64(input_image, mode, weight1, weight2)
 
-    return sotai_image, sketch_image, filename
+    return sotai_image, sketch_image, mixed_image, filename
 
 def mix_images(sotai_image_data, sketch_image_data, opacity1, opacity2):
     sotai_image = Image.open(io.BytesIO(base64.b64decode(sotai_image_data))).convert('RGBA')
@@ -155,12 +155,12 @@ with gr.Blocks() as demo:
         original_submit.click(
             process_image,
             inputs=[input_image, original_mode],
-            outputs=[sotai_image_data, sketch_image_data, send_filename]
+            outputs=[sotai_image_data, sketch_image_data, mixed_image, send_filename]
         )
         refine_submit.click(
             process_image,
             inputs=[input_image, refine_input[0], refine_input[1], refine_input[2]],
-            outputs=[sotai_image_data, sketch_image_data, send_filename]
+            outputs=[sotai_image_data, sketch_image_data, mixed_image, send_filename]
         )
         sotai_image_data.change(
             mix_images,