Update app.py

app.py

@@ -15,6 +15,7 @@ import torch
 from diffusers import StableDiffusionPipeline, StableDiffusionControlNetInpaintPipeline, ControlNetModel
 from PIL import Image
 from compel import Compel
+from diffusers import EulerDiscreteScheduler
 
 # Device configuration
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -40,6 +41,10 @@ pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
     torch_dtype=torch.float16,
 ).to(device)
 
+
+# Set the K_EULER scheduler
+pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
+
 # Function to segment hair and generate mask
 def segment_hair(image):
     rgba_image = cv2.cvtColor(image, cv2.COLOR_BGR2RGBA)
@@ -89,7 +94,7 @@ def inpaint_hair(image, prompt):
         image=image_pil,
         mask_image=mask_pil,
         control_image=inpaint_condition,
-        num_inference_steps=50,
+        num_inference_steps=25,
         guidance_scale=7.5,
         generator=generator
     ).images[0]