sand74 commited on
Commit
57dd83c
·
verified ·
1 Parent(s): 2ce7c8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -101,6 +101,17 @@ examples = [
101
  "changpu_hw walking in field",
102
  ]
103
 
 
 
 
 
 
 
 
 
 
 
 
104
  css = """
105
  #col-container {
106
  margin: 0 auto;
@@ -160,7 +171,7 @@ with gr.Blocks(css=css) as demo:
160
  interactive=True
161
  )
162
  controlnet_mode = gr.Dropdown(
163
- ["edge_detection", "pose_estimation", "depth_map", "scribble"],
164
  value="edge_detection",
165
  label="ControlNet mode",
166
  interactive=True
 
101
  "changpu_hw walking in field",
102
  ]
103
 
104
+ control_net_modes = [
105
+ "canny edge detection",
106
+ "Midas depth estimation",
107
+ "HED edge detection (soft edge)",
108
+ "M-LSD line detection",
109
+ "normal map",
110
+ "OpenPose pose detection",
111
+ "human scribbles",
112
+ "semantic segmentation"
113
+ ]
114
+
115
  css = """
116
  #col-container {
117
  margin: 0 auto;
 
171
  interactive=True
172
  )
173
  controlnet_mode = gr.Dropdown(
174
+ control_net_modes,
175
  value="edge_detection",
176
  label="ControlNet mode",
177
  interactive=True