diff --git a/apps/stable_diffusion/web/ui/txt2img_ui.py b/apps/stable_diffusion/web/ui/txt2img_ui.py
index 731c8a414b..7e90d314d3 100644
--- a/apps/stable_diffusion/web/ui/txt2img_ui.py
+++ b/apps/stable_diffusion/web/ui/txt2img_ui.py
@@ -314,7 +314,6 @@ def resource_path(relative_path):
 
 with gr.Blocks(title="Text-to-Image", css=dark_theme) as txt2img_web:
     with gr.Row(elem_id="ui_title"):
-        breakpoint()
         nod_logo = Image.open(nodlogo_loc)
         with gr.Row():
             with gr.Column(scale=1, elem_id="demo_title_outer"):
diff --git a/apps/stable_diffusion/web/ui/upscaler_ui.py b/apps/stable_diffusion/web/ui/upscaler_ui.py
index 7f482a1ef1..318453f4d8 100644
--- a/apps/stable_diffusion/web/ui/upscaler_ui.py
+++ b/apps/stable_diffusion/web/ui/upscaler_ui.py
@@ -311,7 +311,7 @@ def upscaler_inf(
                         elem_id="negative_prompt_box",
                     )
 
-                upscaler_init_image = gr.ImageEditor(
+                upscaler_init_image = gr.Image(
                     label="Input Image",
                     type="pil",
                     height=300,