Update app.py

app.py

@@ -78,7 +78,7 @@ def process_image_and_text(subject_image, adapter_dict, prompt, _unused1, _unuse
         try:
             blended_bytes = base64.b64decode(data["blended"])
             raw_img = Image.open(BytesIO(blended_bytes)).convert("RGB")
-            return [[raw_img]], raw_img
+            return raw_img, raw_img
         except Exception:
             raise gr.Error("Failed to decode 'blended' image from Baseten response.")
     else:
@@ -128,7 +128,7 @@ with gr.Blocks(css=css, title="ZenCtrl Inpainting") as demo:
                 bg_btn = gr.Button("Generate BG")
 
         with gr.Column(scale=2):
-            gallery = gr.Gallery(columns=[1], rows=[1], object_fit="contain", height="auto")
+            output_img = gr.Image(label="Output Image")
             bg_img = gr.Image(label="Background", visible=True)
 
     # ---------- Example wrapper ---------------------------------
@@ -159,10 +159,15 @@ with gr.Blocks(css=css, title="ZenCtrl Inpainting") as demo:
     )
 
     # ---------- Buttons & interactions --------------------------
+    # run_btn.click(
+    #     process_image_and_text,
+    #     inputs=[subj_img, ref_img, promptbox, gr.State(False), gr.State("")],
+    #     outputs=[gallery, raw_state]
+    # )
     run_btn.click(
         process_image_and_text,
         inputs=[subj_img, ref_img, promptbox, gr.State(False), gr.State("")],
-        outputs=[gallery, raw_state]
+        outputs=[output_img, raw_state]
     )
     bg_btn.click(_gen_bg, inputs=[bgprompt], outputs=[bg_img])