Rename app_canny.py to app_edge.py

app_canny.py → app_edge.py

@@ -18,10 +18,10 @@ examples = [
         "condition/example/t2i/multigen/girl.jpg",
         "An anime style girl with blue hair", "(512, 512)"
     ],
-    [
-        "condition/example/t2i/multi_resolution/bird.jpg", "colorful bird",
-        "(921, 564)"
-    ],
+    # [
+    #     "condition/example/t2i/multi_resolution/bird.jpg", "colorful bird",
+    #     "(921, 564)"
+    # ],
 ]
 
 
@@ -33,6 +33,18 @@ def create_demo(process):
                 prompt = gr.Textbox(label="Prompt")
                 run_button = gr.Button("Run")
                 with gr.Accordion("Advanced options", open=False):
+                    preprocessor_name = gr.Radio(
+                        label="Preprocessor",
+                        choices=[
+                            "Canny",
+                            "Hed",
+                            "Lineart",
+                            "No preprocess",
+                        ],
+                        type="value",
+                        value="Canny",
+                        info='Edge type.',
+                    )
                     canny_low_threshold = gr.Slider(
                         label="Canny low threshold",
                         minimum=0,
@@ -50,6 +62,7 @@ def create_demo(process):
                                           maximum=30.0,
                                           value=2,
                                           step=0.1)
+                    control_strength = gr.Slider(minimum=0., maximum=1.0, step=0.1, value=1.0, label="control_strength")
                     relolution = gr.Slider(label="(H, W)",
                                            minimum=384,
                                            maximum=768,
@@ -101,6 +114,8 @@ def create_demo(process):
             seed,
             canny_low_threshold,
             canny_high_threshold,
+            control_strength,
+            preprocessor_name,
         ]
         # prompt.submit(
         #     fn=randomize_seed_fn,
@@ -124,7 +139,7 @@ def create_demo(process):
             fn=process,
             inputs=inputs,
             outputs=result,
-            api_name="canny",
+            api_name="edge",
         )
     return demo
 
@@ -132,5 +147,5 @@ def create_demo(process):
 if __name__ == "__main__":
     from model import Model
     model = Model()
-    demo = create_demo(model.process_canny)
+    demo = create_demo(model.process_edge)
     demo.queue().launch(share=False, server_name="0.0.0.0")