Update app.py

app.py

@@ -18,7 +18,7 @@ DEVICE = "cuda" #if torch.cuda.is_available() else "cpu"
 pipe = FluxInpaintPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(DEVICE)
 
 @spaces.GPU()
-def process(input_image_editor, input_text, strength, seed, randomize_seed, num_inference_steps, guidance_scale=3.5, progress=gr.Progress(track_tqdm=True)):
+def process(input_image_editor, mask_image, input_text, strength, seed, randomize_seed, num_inference_steps, guidance_scale=3.5, progress=gr.Progress(track_tqdm=True)):
     if not input_text:
         raise gr.Error("Please enter a text prompt.")
 
@@ -100,7 +100,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     submit_button_component.click(
         fn=process,
-        inputs=[input_image_component, input_text_component, strength_slider, seed_number, randomize_seed, num_inference_steps, guidance_scale],
+        inputs=[input_image_component, uploaded_mask_component, input_text_component, strength_slider, seed_number, randomize_seed, num_inference_steps, guidance_scale],
         outputs=[output_image_component, output_mask_component, output_seed]
     )