inpaint

app.py

@@ -26,10 +26,10 @@ base_model = "black-forest-labs/FLUX.1-dev"
 controlnet_model = "InstantX/FLUX.1-dev-Controlnet-Canny"
 
 controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=dtype)
-pipe = FluxControlNetInpaintPipeline.from_pretrained(
-    base_model, controlnet=controlnet, torch_dtype=dtype
-).to(device)
-# pipe = FluxInpaintPipeline.from_pretrained(base_model, torch_dtype=dtype).to(device)
+# pipe = FluxControlNetInpaintPipeline.from_pretrained(
+#     base_model, controlnet=controlnet, torch_dtype=dtype
+# ).to(device)
+pipe = FluxInpaintPipeline.from_pretrained(base_model, torch_dtype=dtype).to(device)
 
 pipe.enable_model_cpu_offload()
 
@@ -93,10 +93,9 @@ def inpaint(
     resized_mask = mask.resize((width, height), Image.LANCZOS)
 
     image_res = pipe(
-        prompt="red l-shaped sofa, red stripes, sleeping cat",
-        prompt_2="red l-shaped sofa, red stripes, sleeping cat.",
+        prompt,
         image=resized_image,
-        control_image=canny_image,
+        # control_image=canny_image,
         mask_image=resized_mask,
         strength=strength,
         num_inference_steps=num_inference_steps,