Update app.py

app.py

@@ -59,15 +59,16 @@ pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
 
 pipe.to("cuda")
 
-# inpaint_model = hf_hub_download(
-#     "andro-flock/LUSTIFY-SDXL-NSFW-checkpoint-v2-0-INPAINTING",
-#     "lustifySDXLNSFW_v20-inpainting.safetensors",
-# )
-pipe_inpaint = StableDiffusionXLFillPipeline.from_pretrained(
+inpaint_model = hf_hub_download(
     "andro-flock/LUSTIFY-SDXL-NSFW-checkpoint-v2-0-INPAINTING",
+    "lustifySDXLNSFW_v20-inpainting.safetensors",
+)
+pipe_inpaint = StableDiffusionXLFillPipeline.from_pretrained(
+    inpaint_model,
     torch_dtype=torch.float16,
     vae=vae,
     controlnet=model,
+    use_safetensors=True
 )
 pipe_inpaint.to("cuda")