kadirnar commited on
Commit
3f7d745
1 Parent(s): df1dcc9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +50 -45
app.py CHANGED
@@ -39,7 +39,7 @@ stable_negative_prompt_list = [
39
  ]
40
  app = gr.Blocks()
41
  with app:
42
- gr.Markdown("# **<h2 align='center'>Stable Diffusion + ControlNet WebUI<h2>**")
43
  gr.Markdown(
44
  """
45
  <h5 style='text-align: center'>
@@ -196,8 +196,8 @@ with app:
196
  controlnet_image_file = gr.Image(label='Image')
197
 
198
  controlnet_model_id = gr.Dropdown(
199
- choices=stable_inpiant_model_list,
200
- value=stable_inpiant_model_list[0],
201
  label='Stable Model Id'
202
  )
203
 
@@ -520,12 +520,12 @@ with app:
520
  controlnet_canny_predict.click(
521
  fn = stable_diffusion_controlnet_img2img,
522
  inputs = [
523
- inpaint_image_file,
524
- inpaint_model_id,
525
- inpaint_prompt,
526
- inpaint_negative_prompt,
527
- inpaint_guidance_scale,
528
- inpaint_num_inference_step,
529
  ],
530
  outputs = [output_image],
531
  )
@@ -533,72 +533,77 @@ with app:
533
  controlnet_hed_predict.click(
534
  fn = stable_diffusion_controlnet_img2img,
535
  inputs = [
536
- inpaint_image_file,
537
- inpaint_model_id,
538
- inpaint_prompt,
539
- inpaint_negative_prompt,
540
- inpaint_guidance_scale,
541
- inpaint_num_inference_step,
542
  ],
543
  outputs = [output_image],
544
  )
 
545
  controlnet_mlsd_predict.click(
546
  fn = stable_diffusion_controlnet_img2img,
547
  inputs = [
548
- inpaint_image_file,
549
- inpaint_model_id,
550
- inpaint_prompt,
551
- inpaint_negative_prompt,
552
- inpaint_guidance_scale,
553
- inpaint_num_inference_step,
554
  ],
555
  outputs = [output_image],
556
  )
 
557
  controlnet_seg_predict.click(
558
  fn = stable_diffusion_controlnet_img2img,
559
  inputs = [
560
- inpaint_image_file,
561
- inpaint_model_id,
562
- inpaint_prompt,
563
- inpaint_negative_prompt,
564
- inpaint_guidance_scale,
565
- inpaint_num_inference_step,
566
  ],
567
  outputs = [output_image],
568
  )
 
569
  controlnet_depth_predict.click(
570
  fn = stable_diffusion_controlnet_img2img,
571
  inputs = [
572
- inpaint_image_file,
573
- inpaint_model_id,
574
- inpaint_prompt,
575
- inpaint_negative_prompt,
576
- inpaint_guidance_scale,
577
- inpaint_num_inference_step,
578
  ],
579
  outputs = [output_image],
580
  )
 
581
  controlnet_scribble_predict.click(
582
  fn = stable_diffusion_controlnet_img2img,
583
  inputs = [
584
- inpaint_image_file,
585
- inpaint_model_id,
586
- inpaint_prompt,
587
- inpaint_negative_prompt,
588
- inpaint_guidance_scale,
589
- inpaint_num_inference_step,
590
  ],
591
  outputs = [output_image],
592
  )
 
593
  controlnet_pose_predict.click(
594
  fn = stable_diffusion_controlnet_img2img,
595
  inputs = [
596
- inpaint_image_file,
597
- inpaint_model_id,
598
- inpaint_prompt,
599
- inpaint_negative_prompt,
600
- inpaint_guidance_scale,
601
- inpaint_num_inference_step,
602
  ],
603
  outputs = [output_image],
604
  )
 
39
  ]
40
  app = gr.Blocks()
41
  with app:
42
+ gr.Markdown("# **<h2 align='center'>Stable Diffusion WebUI<h2>**")
43
  gr.Markdown(
44
  """
45
  <h5 style='text-align: center'>
 
196
  controlnet_image_file = gr.Image(label='Image')
197
 
198
  controlnet_model_id = gr.Dropdown(
199
+ choices=stable_model_list,
200
+ value=stable_model_list[0],
201
  label='Stable Model Id'
202
  )
203
 
 
520
  controlnet_canny_predict.click(
521
  fn = stable_diffusion_controlnet_img2img,
522
  inputs = [
523
+ controlnet_image_file,
524
+ controlnet_model_id,
525
+ controlnet_prompt,
526
+ controlnet_negative_prompt,
527
+ controlnet_guidance_scale,
528
+ controlnet_num_inference_step,
529
  ],
530
  outputs = [output_image],
531
  )
 
533
  controlnet_hed_predict.click(
534
  fn = stable_diffusion_controlnet_img2img,
535
  inputs = [
536
+ controlnet_image_file,
537
+ controlnet_model_id,
538
+ controlnet_prompt,
539
+ controlnet_negative_prompt,
540
+ controlnet_guidance_scale,
541
+ controlnet_num_inference_step,
542
  ],
543
  outputs = [output_image],
544
  )
545
+
546
  controlnet_mlsd_predict.click(
547
  fn = stable_diffusion_controlnet_img2img,
548
  inputs = [
549
+ controlnet_image_file,
550
+ controlnet_model_id,
551
+ controlnet_prompt,
552
+ controlnet_negative_prompt,
553
+ controlnet_guidance_scale,
554
+ controlnet_num_inference_step,
555
  ],
556
  outputs = [output_image],
557
  )
558
+
559
  controlnet_seg_predict.click(
560
  fn = stable_diffusion_controlnet_img2img,
561
  inputs = [
562
+ controlnet_image_file,
563
+ controlnet_model_id,
564
+ controlnet_prompt,
565
+ controlnet_negative_prompt,
566
+ controlnet_guidance_scale,
567
+ controlnet_num_inference_step,
568
  ],
569
  outputs = [output_image],
570
  )
571
+
572
  controlnet_depth_predict.click(
573
  fn = stable_diffusion_controlnet_img2img,
574
  inputs = [
575
+ controlnet_image_file,
576
+ controlnet_model_id,
577
+ controlnet_prompt,
578
+ controlnet_negative_prompt,
579
+ controlnet_guidance_scale,
580
+ controlnet_num_inference_step,
581
  ],
582
  outputs = [output_image],
583
  )
584
+
585
  controlnet_scribble_predict.click(
586
  fn = stable_diffusion_controlnet_img2img,
587
  inputs = [
588
+ controlnet_image_file,
589
+ controlnet_model_id,
590
+ controlnet_prompt,
591
+ controlnet_negative_prompt,
592
+ controlnet_guidance_scale,
593
+ controlnet_num_inference_step,
594
  ],
595
  outputs = [output_image],
596
  )
597
+
598
  controlnet_pose_predict.click(
599
  fn = stable_diffusion_controlnet_img2img,
600
  inputs = [
601
+ controlnet_image_file,
602
+ controlnet_model_id,
603
+ controlnet_prompt,
604
+ controlnet_negative_prompt,
605
+ controlnet_guidance_scale,
606
+ controlnet_num_inference_step,
607
  ],
608
  outputs = [output_image],
609
  )