Update app.py

app.py

@@ -259,7 +259,7 @@ def create_demo(model):
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(label="Input Image", type='pil', elem_id='img-display-input')
-            input_text = gr.Textbox(label='Prompt', value="masterpiece, high quality, ", lines=2)
+            input_text = gr.Textbox(label='Prompt', value="high resolution, clay render style , grayscale", lines=2)
             with gr.Accordion('Advanced options', open=False):
                 num_steps = gr.Slider(label='Steps',
                                       minimum=1,
@@ -283,21 +283,22 @@ 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")
+                    value="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():
-            design_gallery = gr.Gallery(label="Output Gallery", elem_id='img-display-output', elem_id_css={'height': '600px', 'width': '800px'})
-
+            design_image = gr.Image(label="Output Mask", elem_id='img-display-output')
+    
+    
     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
@@ -306,12 +307,13 @@ 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 a list containing the generated image
+        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)
+    submit.click(on_submit, inputs=[input_image, input_text, num_steps, guidance_scale, seed, strength, a_prompt, n_prompt, img_size], outputs=design_image)
     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)
 controlnet_seg = ControlNetModel.from_pretrained(
@@ -343,7 +345,7 @@ def main():
     model = ControlNetDepthDesignModelMulti()
     print('Models uploaded successfully')
     
-    title = "# StableDesign by TDN-M"
+    title = "# StableDesign"
     description = """
     WELCOME
     """