Fix image return types.

nod-ai · Dec 2, 2023 · 0afc605 · 0afc605
1 parent 15edeab
commit 0afc605
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 4 deletions.
diff --git a/apps/stable_diffusion/web/ui/inpaint_ui.py b/apps/stable_diffusion/web/ui/inpaint_ui.py
@@ -288,7 +288,7 @@ def inpaint_inf(
                         elem_id="negative_prompt_box",
                     )
 
-                inpaint_init_image = gr.ImageEditor(
+                inpaint_init_image = gr.Image(
                     label="Masked Image",
                     sources="upload",
                     type="pil",

diff --git a/apps/stable_diffusion/web/ui/outpaint_ui.py b/apps/stable_diffusion/web/ui/outpaint_ui.py
@@ -291,7 +291,7 @@ def outpaint_inf(
                         elem_id="negative_prompt_box",
                     )
 
-                outpaint_init_image = gr.ImageEditor(
+                outpaint_init_image = gr.Image(
                     label="Input Image",
                     type="pil",
                     height=300,

diff --git 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
@@ -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,