Update app.py

app.py

@@ -448,7 +448,51 @@ class ControlNetDepthDesignModelMulti:
         
         return design_image
 
-
+def create_demo(model):
+    gr.Markdown("### Just try zeroGPU")
+    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', placeholder='Please upload your image first', lines=2)
+            with gr.Accordion('Advanced options', open=False):
+                num_steps = gr.Slider(label='Steps',
+                                      minimum=1,
+                                      maximum=50,
+                                      value=50,
+                                      step=1)
+                img_size = gr.Slider(label='Image size',
+                                      minimum=256,
+                                      maximum=768,
+                                      value=768,
+                                      step=64)
+                guidance_scale = gr.Slider(label='Guidance Scale',
+                                           minimum=0.1,
+                                           maximum=30.0,
+                                           value=10.0,
+                                           step=0.1)
+                seed = gr.Slider(label='Seed',
+                                 minimum=-1,
+                                 maximum=2147483647,
+                                 value=323*111,
+                                 step=1,
+                                 randomize=True)
+                strength = gr.Slider(label='Strength',
+                                           minimum=0.1,
+                                           maximum=1.0,
+                                           value=0.9,
+                                           step=0.1)
+                a_prompt = gr.Textbox(
+                    label='Added Prompt',
+                    value="interior design, 4K, high resolution, photorealistic")
+                n_prompt = gr.Textbox(
+                    label='Negative Prompt',
+                    value="window, door, low resolution, banner, logo, watermark, text, deformed, blurry, out of focus, surreal, ugly, beginner")
+            submit = gr.Button("Submit")
+        
+        with gr.Column():
+            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
@@ -460,10 +504,7 @@ class ControlNetDepthDesignModelMulti:
         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_image)
-    examples = gr.Examples(examples=[["imgs/bedroom_1.jpg", "An elegantly appointed bedroom in the Art Deco style, featuring a grand king-size bed with geometric bedding, a luxurious velvet armchair, and a mirrored nightstand that reflects the room's opulence. Art Deco-inspired artwork adds a touch of glamour"], ["imgs/bedroom_2.jpg", "A bedroom that exudes French country charm with a soft upholstered bed, walls adorned with floral wallpaper, and a vintage wooden wardrobe. A crystal chandelier casts a warm, inviting glow over the space"], ["imgs/dinning_room_1.jpg", "A cozy dining room that captures the essence of rustic charm with a solid wooden farmhouse table at its core, surrounded by an eclectic mix of mismatched chairs. An antique sideboard serves as a statement piece, and the ambiance is warmly lit by a series of quaint Edison bulbs dangling from the ceiling"], ["imgs/dinning_room_3.jpg", "A dining room that epitomizes contemporary elegance, anchored by a sleek, minimalist dining table paired with stylish modern chairs. Artistic lighting fixtures create a focal point above, while the surrounding minimalist decor ensures the space feels open, airy, and utterly modern"], ["imgs/image_1.jpg", "A glamorous master bedroom in Hollywood Regency style, boasting a plush tufted headboard, mirrored furniture reflecting elegance, luxurious fabrics in rich textures, and opulent gold accents for a touch of luxury."], ["imgs/image_2.jpg", "A vibrant living room with a tropical theme, complete with comfortable rattan furniture, large leafy plants bringing the outdoors in, bright cushions adding pops of color, and bamboo blinds for natural light control."], ["imgs/living_room_1.jpg", "A stylish living room embracing mid-century modern aesthetics, featuring a vintage teak coffee table at its center, complemented by a classic sunburst clock on the wall and a cozy shag rug underfoot, creating a warm and inviting atmosphere"]],
-                           inputs=[input_image, input_text], cache_examples=False)
-
-
+   
 
 controlnet_depth= ControlNetModel.from_pretrained(
     "controlnet_depth", torch_dtype=dtype, use_safetensors=True)