22

app.py

@@ -80,7 +80,7 @@ pipe.to("cuda", torch.bfloat16)
 print("pipe ready.")
 
 def prepare(edit_dict):
-    return edit_dict['composite']
+    return edit_dict['layers'][0]
 
 @spaces.GPU
 def inference(
@@ -254,7 +254,7 @@ with gr.Blocks() as demo:
                 output_image = gr.Image(label="Generated image", show_label=False)
                 # polished_prompt = gr.Textbox(label="Polished prompt", interactive=False)
 
-                with gr.Accordion("Preprocessor data", open=False):
+                with gr.Accordion("Preprocessor data", open=True):
                     mask_image = gr.Image(label="Mask image", show_label=False)
                     
         gr.Examples(examples=examples, inputs=[edit_dict, prompt])