add condition

app.py

@@ -61,7 +61,7 @@ with demo_blocks as demo:
         gr.HTML(read_file("demo_header.html"))
     with gr.Row():
                 with gr.Column():
-                    image = gr.ImageEditor(height=1000,sources=['upload','clipboard'],transforms=[],image_mode='RGB', layers=False,  elem_id="image_upload", type="pil", label="Upload",brush=gr.Brush(colors=["#999"], color_mode="fixed"))
+                    image = gr.ImageEditor(height=1000,sources=['upload','clipboard'],transforms=[],image_mode='RGB', layers=False,  elem_id="image_upload", type="pil", label="Upload",brush=gr.Brush(colors=["#fff"], color_mode="fixed"))
                     with gr.Row(elem_id="prompt-container",  equal_height=False):
                         with gr.Row():
                             prompt = gr.Textbox(label="Prompt",value="a eyes closed girl,shut eyes",placeholder="Your prompt (what you want in place of what is erased)", elem_id="prompt")

flux1_inpaint.py

@@ -36,12 +36,13 @@ def make_inpaint_condition(image, image_mask):
 def process_image(image,mask_image,prompt="a girl",negative_prompt="",model_id="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0,num_inference_steps=4):
     
     
-
-    #control_image=make_inpaint_condition(image,mask_image)
-    #image.save("_control.jpg")
     if image == None:
         return None
     
+    control_image=make_inpaint_condition(image,mask_image)
+    #image.save("_control.jpg")
+    image = control_image
+    
     pipe = FluxInpaintPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
     pipe.to(device)
     #batch_size =1