init

demo_header.html

@@ -1,13 +1,12 @@
 <div style="text-align: center;">
     <h1>
-        Flux.1-schnell mask-inpaint
+        Flux.1-schnell mask-img2img
     </h1>
     <p>Making face images for <a href="https://huggingface.co/spaces/Akjava/godot-huggingface-chain">AI Diagram Chat with Voice/Face Character</a></p>
     <p>
         
         See examples.
-        The image requires guide painting.(To generate an image with closed eyes, fill the eye area with skin color.)
-        You can draw mask or upload mask.BE CAREFUL UPLOAEDED MASK(If the uploaded mask exist,ignore your painted mask)
+        
     </p>
-    <p>Hint:step by step,reduce strength.blur guide-paint and mask</p>
+    <p>Hint:</p>
 </div>

flux1_img2img.py

@@ -1,5 +1,5 @@
 import torch
-from diffusers import FluxInpaintPipeline
+from diffusers import FluxImg2ImgPipeline
 
 from PIL import Image
 import sys
@@ -12,15 +12,16 @@ def process_image(image,mask_image,prompt="a person",model_id="black-forest-labs
     if image == None:
         return None
     
-    pipe = FluxInpaintPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
+    pipe = FluxImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
     pipe.to("cuda")
 
     generators = []
     generator = torch.Generator("cuda").manual_seed(seed)
     generators.append(generator)
     # more parameter see https://huggingface.co/docs/diffusers/api/pipelines/flux#diffusers.FluxInpaintPipeline
-    output = pipe(prompt=prompt, image=image, mask_image=mask_image,generator=generator,strength=strength)
+    output = pipe(prompt=prompt, image=image,generator=generator,strength=strength)
 
+    # TODO support mask
     return output.images[0]
 
 if __name__ == "__main__":