ok

app.py

@@ -283,16 +283,16 @@ def create_demo(model):
                                  step=1,
                                  randomize=True)
                 strength = gr.Slider(label='Strength',
-                                           minimum=0.1,
-                                           maximum=1.0,
-                                           value=0.9,
-                                           step=0.1)
+                                     minimum=0.1,
+                                     maximum=1.0,
+                                     value=0.9,
+                                     step=0.1)
                 a_prompt = gr.Textbox(
                     label='Added Prompt',
                     value="high resolution, clay render style , grayscale, 8K, high resolution, photorealistic")
                 n_prompt = gr.Textbox(
                     label='Negative Prompt',
-                    value=" low resolution, banner, logo, watermark, deformed, blurry, out of focus, surreal, ugly, beginner")
+                    value="low resolution, banner, logo, watermark, deformed, blurry, out of focus, surreal, ugly, beginner")
             submit = gr.Button("Submit")
         
         with gr.Column():
@@ -306,13 +306,11 @@ def create_demo(model):
         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
+        return [out_img]  # Return a list containing the generated image
 
     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)
-    
-
 
 controlnet_depth= ControlNetModel.from_pretrained(
     "controlnet_depth", torch_dtype=dtype, use_safetensors=True)