Update app.py

app.py

@@ -1,7 +1,7 @@
 import torch
 import spaces
 import gradio as gr
-from diffusers import FluxInpaintPipeline, FluxTransformer2DModel
+from diffusers import FluxFillPipeline
 import random
 import os
 import numpy as np
@@ -9,30 +9,17 @@ from huggingface_hub import hf_hub_download
 
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 MAX_SEED = np.iinfo(np.int32).max
-model = "black-forest-labs/FLUX.1-dev"
-hf_hub_download(repo_id="black-forest-labs/FLUX.1-Fill-dev", filename="ae.safetensors", local_dir=".")
+
 
 if torch.cuda.is_available():
-    transformer = FluxTransformer2DModel.from_single_file(
-        "https://huggingface.co/black-forest-labs/FLUX.1-Fill-dev/blob/main/flux1-fill-dev.safetensors",
-        low_cpu_mem_usage=False,
-        ignore_mismatched_sizes=True,
-        torch_dtype=torch.bfloat16
-    )
-    vae = AutoencoderKL.from_pretrained("./ae.safetensors")
-    pipe = FluxInpaintPipeline.from_pretrained(
-        model,
-        vae=vae,
-        transformer=transformer,
-        torch_dtype=torch.bfloat16)
-    pipe.to("cuda")
+    repo_id = "black-forest-labs/FLUX.1-Fill-dev"
+    pipe = FluxFillPipeline.from_pretrained(repo_id, torch_dtype=torch.bfloat16).to("cuda")
 
 
 @spaces.GPU()
 def inpaintGen(
         imgMask,
         inpaint_prompt: str,
-        strength: float,
         guidance: float,
         num_steps: int,
         seed: int,
@@ -52,7 +39,7 @@ def inpaintGen(
 
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
-    generator = torch.Generator(device=DEVICE).manual_seed(seed)
+    generator = torch.Generator("cpu").manual_seed(seed)
 
     result = pipe(
         prompt=inpaint_prompt,
@@ -61,10 +48,10 @@ def inpaintGen(
         mask_image=mask_img,
         width=width,
         height=height,
-        strength=strength,
         num_inference_steps=num_steps,
         generator=generator,
-        guidance_scale=guidance
+        guidance_scale=guidance,
+        max_sequence_length=512,
     ).images[0]
     
     return result
@@ -88,7 +75,6 @@ with gr.Blocks(theme="ocean", title="Flux.1 dev inpaint", css=CSS) as demo:
                 Inpaint_clearBtn = gr.ClearButton([imgMask, inpaint_prompt], value="Clear")
         image_out = gr.Image(type="pil", label="Output", height=960)
     with gr.Accordion("Advanced ⚙️", open=False):
-        strength = gr.Slider(label="Strength", minimum=0, maximum=1, value=1, step=0.1)
         guidance = gr.Slider(label="Guidance scale", minimum=1, maximum=20, value=7.5, step=0.1)
         num_steps = gr.Slider(label="Steps", minimum=1, maximum=20, value=20, step=1)
         seed = gr.Number(label="Seed", value=42, precision=0)
@@ -103,7 +89,6 @@ with gr.Blocks(theme="ocean", title="Flux.1 dev inpaint", css=CSS) as demo:
         inputs = [
             imgMask,
             inpaint_prompt,
-            strength,
             guidance,
             num_steps,
             seed,