22

app.py

@@ -107,7 +107,7 @@ def inference(
     processor_id = 'canny'
     if control_mode == 'HED':
         processor_id = 'softedge_hed'
-    if control_mode =='Midas':
+    if control_mode =='Depth':
         processor_id = 'depth_midas'
     if control_mode =='MLSD':
         processor_id = 'mlsd'
@@ -184,7 +184,7 @@ with gr.Blocks(css=css) as demo:
                 )
 
                 control_mode = gr.Radio(
-                    choices=["HED", "Canny", "Midas", "MLSD", "Pose"],
+                    choices=["Canny", "Depth", "HED", "MLSD", "Pose"],
                     value="HED",
                     label="Control Mode"
                 )
@@ -240,7 +240,7 @@ with gr.Blocks(css=css) as demo:
                         step=1,
                         value=8,
                     )
-        gr.Examples(examples=examples, inputs=[input_image, prompt])
+        gr.Examples(examples=examples, inputs=[input_image, prompt, control_mode])
         gr.HTML(read_file("static/footer.html"))
 
     run_button.click(

static/data.json

@@ -1,9 +1,9 @@
 {
 	"examples": [
-		["examples/bird_canny.png", "A hummingbird, standing upright on a branch, looking directly at the viewer, high-detail feathers, soft studio lighting."],
-		["examples/hed.jpg", "A middle-aged man with a short beard, wearing a casual button-down shirt, sitting at a polished dark wooden table, holding a tumbler of whiskey with ice and taking a thoughtful sip. The background is a softly lit."],
-		["examples/depth.jpg", "An industrial-style corridor featuring a minimalist table and a sleek sofa, with exposed brick walls and concrete flooring, bathed in natural light from large windows."],
-		["examples/pose.jpg", "A beautiful female paladin, with long flowing blonde hair, stands defiantly in a full suit of gleaming silver plate armor, a holy symbol emblazoned on her shield, in a dramatic graphic novel art style with bold lines and strong chiaroscuro."],
-		["examples/pose2.jpg", "A fit, athletic young woman, squatting low, glancing confidently at the camera. She's on a picturesque tropical beach with gentle waves lapping the shore. The image has the crisp, high-contrast look of a fashion magazine cover. Dynamic pose, bright and inviting."]
+		["examples/pose1.jpg", "A female paladin.", "Pose"],
+		["examples/bird.png", "A bird sitting on a branch", "Canny"],
+		["examples/bottle.png", "A man holding glass", "HED"],
+		["examples/room.png", "modern architecture, living room", "Depth"],
+		["examples/pose2.jpg", "A fit, athletic young woman. beach background", "Pose"]
 	]
 }