TDN-M commited on
Commit
2aa88fe
1 Parent(s): 312ce40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -180,8 +180,8 @@ class ControlNetDepthDesignModelMulti:
180
 
181
  self.seed = 323*111
182
  self.neg_prompt = "window, door, low resolution, banner, logo, watermark, text, deformed, blurry, out of focus, surreal, ugly, beginner"
183
- self.control_items = ["windowpane;window", "door;double;door"]
184
- self.additional_quality_suffix = "interior design, 4K, high resolution, photorealistic"
185
 
186
  @spaces.GPU
187
  def generate_design(self, empty_room_image: Image, prompt: str, guidance_scale: int = 10, num_steps: int = 50, strength: float =0.9, img_size: int = 640) -> Image:
@@ -259,7 +259,7 @@ def create_demo(model):
259
  with gr.Row():
260
  with gr.Column():
261
  input_image = gr.Image(label="Input Image", type='pil', elem_id='img-display-input')
262
- input_text = gr.Textbox(label='Prompt', placeholder='Please upload your image first', lines=2)
263
  with gr.Accordion('Advanced options', open=False):
264
  num_steps = gr.Slider(label='Steps',
265
  minimum=1,
@@ -289,7 +289,7 @@ def create_demo(model):
289
  step=0.1)
290
  a_prompt = gr.Textbox(
291
  label='Added Prompt',
292
- value="4K, high resolution, photorealistic")
293
  n_prompt = gr.Textbox(
294
  label='Negative Prompt',
295
  value=" low resolution, banner, logo, watermark, deformed, blurry, out of focus, surreal, ugly, beginner")
@@ -310,7 +310,7 @@ def create_demo(model):
310
  return out_img
311
 
312
  submit.click(on_submit, inputs=[input_image, input_text, num_steps, guidance_scale, seed, strength, a_prompt, n_prompt, img_size], outputs=design_image)
313
- 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"]],
314
  inputs=[input_image, input_text], cache_examples=False)
315
 
316
 
 
180
 
181
  self.seed = 323*111
182
  self.neg_prompt = "window, door, low resolution, banner, logo, watermark, text, deformed, blurry, out of focus, surreal, ugly, beginner"
183
+ self.control_items = ["ugly", "ugly"]
184
+ self.additional_quality_suffix = "4K, high resolution, photorealistic"
185
 
186
  @spaces.GPU
187
  def generate_design(self, empty_room_image: Image, prompt: str, guidance_scale: int = 10, num_steps: int = 50, strength: float =0.9, img_size: int = 640) -> Image:
 
259
  with gr.Row():
260
  with gr.Column():
261
  input_image = gr.Image(label="Input Image", type='pil', elem_id='img-display-input')
262
+ input_text = gr.Textbox(label='Prompt', value="high resolution, clay render style , grayscale", lines=2)
263
  with gr.Accordion('Advanced options', open=False):
264
  num_steps = gr.Slider(label='Steps',
265
  minimum=1,
 
289
  step=0.1)
290
  a_prompt = gr.Textbox(
291
  label='Added Prompt',
292
+ value="8K, high resolution, photorealistic")
293
  n_prompt = gr.Textbox(
294
  label='Negative Prompt',
295
  value=" low resolution, banner, logo, watermark, deformed, blurry, out of focus, surreal, ugly, beginner")
 
310
  return out_img
311
 
312
  submit.click(on_submit, inputs=[input_image, input_text, num_steps, guidance_scale, seed, strength, a_prompt, n_prompt, img_size], outputs=design_image)
313
+ examples = gr.Examples(examples=[["imgs/bedroom_1.jpg"]],
314
  inputs=[input_image, input_text], cache_examples=False)
315
 
316