Update app.py

app.py

@@ -28,7 +28,7 @@ with gr.Blocks() as demo:
         prompt,
         image
     ]
-    outputs = [gallery]
+    outputs = [gallery_out]
     
     submit_btn.click(infer, inputs, outputs)