change to img2img

app_haircolor_img2img.py

@@ -26,7 +26,7 @@ BASE_MODEL = "stabilityai/stable-diffusion-xl-base-1.0"
 
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
-DEFAULT_EDIT_PROMPT = "change hair to blue"
+DEFAULT_EDIT_PROMPT = "RAW photo, Fujifilm XT3,  sharp hair, high resolution hair, hair tones, natural hair, magazine hair, white color hair"
 
 DEFAULT_CATEGORY = "hair"
 
@@ -94,13 +94,13 @@ def image_to_image(
         generator=generator,
         prompt=edit_prompt,
         image=input_image,
-        control_image=cond_image,
         height=generate_size,
         width=generate_size,
         guidance_scale=guidance_scale,
         strength=strength,
         num_inference_steps=num_steps,
-        controlnet_conditioning_scale=cond_scale,
+        adapter_image=cond_image,
+        adapter_conditioning_scale=cond_scale,
     ).images[0]
     
     run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)