Update app.py

app.py

@@ -7,6 +7,7 @@ import numpy as np
 
 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(
@@ -15,8 +16,10 @@ if torch.cuda.is_available():
         ignore_mismatched_sizes=True,
         torch_dtype=torch.bfloat16
     )
+    vae = AutoencoderKL.from_pretrained("./ae.safetensors")
     pipe = FluxInpaintPipeline.from_pretrained(
-        model, 
+        model,
+        vae=vae,
         transformer=transformer,
         torch_dtype=torch.bfloat16)
     pipe.to("cuda")