22

app.py

@@ -86,6 +86,7 @@ def inference(
     prompt,
     input_image,
     image_scale=1.0,
+    control_mode='Canny'
     control_context_scale = 0.75,
     seed=42,
     randomize_seed=True,
@@ -100,7 +101,7 @@ def inference(
         return None
     
     # input_image, width, height = scale_image(input_image, image_scale)
-    control_mode='HED'
+    # control_mode='HED'
 
     if control_mode == 'HED':
         processor = Processor('softedge_hed')
@@ -177,7 +178,12 @@ with gr.Blocks(css=css) as demo:
                     container=False,
                 )
 
-                run_button = gr.Button("Run", variant="primary")
+                control_mode = gr.Radio(
+                    choices=["HED", "Canny", "Midas", "MLSD", "Pose"],
+                    value="Pose",
+                    label="Control Mode"
+                )
+                run_button = gr.Button("Generate", variant="primary")
             with gr.Column():
                 output_image = gr.Image(label="Generated image", show_label=False)
                 polished_prompt = gr.Textbox(label="Polished prompt", interactive=False)
@@ -246,6 +252,7 @@ with gr.Blocks(css=css) as demo:
             polished_prompt,
             input_image,
             image_scale,
+            control_mode,
             control_context_scale,
             seed,
             randomize_seed,