22

app.py

@@ -242,7 +242,11 @@ with gr.Blocks() as demo:
                 # polished_prompt = gr.Textbox(label="Polished prompt", interactive=False)
 
                 with gr.Accordion("Preprocessor data", open=False):
-                    mask_image = gr.Image(label="Generated Mask", interactive=False)
+                    mask_image = gr.Image(
+                        label="Generated Mask", 
+                        interactive=False,
+                        type="pil",
+                    )
                     
         gr.Examples(examples=examples, inputs=[edit_dict, prompt])
         gr.Markdown(read_file("static/footer.md"))