Update app.py

app.py

@@ -233,9 +233,9 @@ def run(
         image = Image.fromarray(controlnet_img)
     elif not use_hed:
         controlnet_img = resized_image
+        image = resized_image
     else:
         controlnet_img = processor(resized_image, scribble=False)
-        # Process controlnet_img as before...
         controlnet_img = np.array(controlnet_img)
         controlnet_img = nms(controlnet_img, 127, 3)
         controlnet_img = cv2.GaussianBlur(controlnet_img, (0, 0), 3)
@@ -275,7 +275,6 @@ def run(
 
     return (controlnet_img, out)
 
-
 with gr.Blocks(css="style.css", js=js_func) as demo:
     gr.Markdown(DESCRIPTION, elem_id="description")
     gr.DuplicateButton(
@@ -287,7 +286,7 @@ with gr.Blocks(css="style.css", js=js_func) as demo:
     with gr.Row():
         with gr.Column():
             with gr.Group():
-                image = gr.ImageEditor(type="pil",label="Sketch your image or upload one", crop_size="1:1", width=512, height=512,)
+                image = gr.ImageEditor(type="pil", label="Sketch your image or upload one", width=512, height=512)
                 prompt = gr.Textbox(label="Prompt")
                 style = gr.Dropdown(label="Style", choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME)
                 use_hed = gr.Checkbox(label="use HED detector", value=False, info="check this box if you upload an image and want to turn it to a sketch")