RamAnanth1 commited on
Commit
608c551
·
1 Parent(s): 85ef8cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -31,7 +31,7 @@ for name in names:
31
  subprocess.run(shlex.split(command), cwd='ControlNet/annotator/ckpts/')
32
 
33
 
34
- from app_depth import create_demo as create_demo_depth
35
  from model import Model, download_all_controlnet_weights
36
 
37
  DESCRIPTION = '# RoomGPT <p> Redesign your room using the power of AI</p>'
@@ -50,7 +50,7 @@ model = Model(base_model_id=DEFAULT_MODEL_ID)
50
 
51
  with gr.Blocks(css='style.css') as demo:
52
  gr.Markdown(DESCRIPTION)
53
- create_demo_depth(model.process_depth,
54
  max_images=MAX_IMAGES,
55
  default_num_images=DEFAULT_NUM_IMAGES)
56
 
 
31
  subprocess.run(shlex.split(command), cwd='ControlNet/annotator/ckpts/')
32
 
33
 
34
+ from app_hough import create_demo as create_demo_hough
35
  from model import Model, download_all_controlnet_weights
36
 
37
  DESCRIPTION = '# RoomGPT <p> Redesign your room using the power of AI</p>'
 
50
 
51
  with gr.Blocks(css='style.css') as demo:
52
  gr.Markdown(DESCRIPTION)
53
+ create_demo_hough(model.process_hough,
54
  max_images=MAX_IMAGES,
55
  default_num_images=DEFAULT_NUM_IMAGES)
56