Update app.py

app.py

@@ -15,7 +15,7 @@ import numpy as np
 MODELS = {
     "RealVisXL V5.0 Lightning": "SG161222/RealVisXL_V5.0_Lightning",
     "Lustify Lightning": "GraydientPlatformAPI/lustify-lightning",
-    "Lustify Inpaint": "andro-flock/LUSTIFY-SDXL-NSFW-checkpoint-v2-0-INPAINTING",
+    # "Lustify Inpaint": "andro-flock/LUSTIFY-SDXL-NSFW-checkpoint-v2-0-INPAINTING",
     "Juggernaut XL Lightning": "RunDiffusion/Juggernaut-XL-Lightning",
 }
 
@@ -59,18 +59,18 @@ 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_single_file(
-    "https://huggingface.co/andro-flock/LUSTIFY-SDXL-NSFW-checkpoint-v2-0-INPAINTING/raw/main/lustifySDXLNSFW_v20-inpainting.safetensors",
-    torch_dtype=torch.float16,
-    vae=vae,
-    controlnet=model,
-    use_safetensors=True
-)
-pipe_inpaint.to("cuda")
+# inpaint_model = hf_hub_download(
+#     "andro-flock/LUSTIFY-SDXL-NSFW-checkpoint-v2-0-INPAINTING",
+#     "lustifySDXLNSFW_v20-inpainting.safetensors",
+# )
+# pipe_inpaint = StableDiffusionXLFillPipeline.from_single_file(
+#     "https://huggingface.co/andro-flock/LUSTIFY-SDXL-NSFW-checkpoint-v2-0-INPAINTING/raw/main/lustifySDXLNSFW_v20-inpainting.safetensors",
+#     torch_dtype=torch.float16,
+#     vae=vae,
+#     controlnet=model,
+#     use_safetensors=True
+# )
+# pipe_inpaint.to("cuda")
 
 @spaces.GPU(duration=16)
 def fill_image(prompt, image, model_selection, paste_back):