Update app.py

app.py

@@ -151,9 +151,6 @@ def infer(
     controlnet_changed = False
 
     if cn_enable:
-        print(cn_mode)
-        print(dict_controlnet)
-        print(pipe.controlnet._name_or_path)
         if dict_controlnet[cn_mode] != pipe.controlnet._name_or_path:
             controlnet = ControlNetModel.from_pretrained(
                 dict_controlnet[cn_mode], 
@@ -204,6 +201,7 @@ def infer(
         params['controlnet_conditioning_scale'] = cn_strength
         if cn_mode == 'edge_detection':
             control_image = map_edge_detection(cn_image)
+            print(type(control_image))
         elif cn_mode == 'scribble':
             control_image = map_scribble(cn_image)
         params['control_image'] = control_image
@@ -293,6 +291,7 @@ with gr.Blocks(css=css) as demo:
                 cn_strength = gr.Slider(0, 2, value=0.8, step=0.1, label="Control strength", interactive=True)
                 cn_mode = gr.Dropdown(
                     choices=["edge_detection", "scribble"],
+                    value="edge_detection",
                     label="Work regime",
                     interactive=True,
                 )