ok

app.py

@@ -296,7 +296,21 @@ def create_demo(model):
             submit = gr.Button("Submit")
         
         with gr.Column():
-            design_image = gr.Image(label="Output Mask", elem_id='img-display-output')
+            design_gallery = gr.Gallery(label="Output Gallery", elem_id='img-display-output').style(grid=[2], height='auto')
+    
+    def on_submit(image, text, num_steps, guidance_scale, seed, strength, a_prompt, n_prompt, img_size):
+        model.seed = seed
+        model.neg_prompt = n_prompt
+        model.additional_quality_suffix = a_prompt
+        
+        with torch.no_grad():
+            out_img = model.generate_design(image, text, guidance_scale=guidance_scale, num_steps=num_steps, strength=strength, img_size=img_size)
+
+        return out_img
+
+    submit.click(on_submit, inputs=[input_image, input_text, num_steps, guidance_scale, seed, strength, a_prompt, n_prompt, img_size], outputs=design_gallery)
+    examples = gr.Examples(examples=[["imgs/bedroom_1.jpg"]],
+                           inputs=[input_image, input_text], cache_examples=False)
     
     
     def on_submit(image, text, num_steps, guidance_scale, seed, strength, a_prompt, n_prompt, img_size):