Update app.py

app.py

@@ -36,20 +36,20 @@ repo_nsfw_classifier = AutoModelForImageClassification.from_pretrained("Falconsa
 processor_nsfw_classifier = ViTImageProcessor.from_pretrained("Falconsai/nsfw_image_detection")
 
 repo_default = StableDiffusionXLPipeline.from_pretrained("fluently/Fluently-XL-Final", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
-repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
-repo_default.set_adapters(["base"], adapter_weights=[0.7])
+#repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
+#repo_default.set_adapters(["base"], adapter_weights=[0.7])
 
-repo_pixel = StableDiffusionXLPipeline.from_pretrained("fluently/Fluently-XL-Final", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
-repo_pixel.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="base")
-repo_pixel.load_lora_weights("nerijs/pixel-art-xl", adapter_name="base2")
-repo_pixel.set_adapters(["base", "base2"], adapter_weights=[1, 1])
+repo_pixel = None#StableDiffusionXLPipeline.from_pretrained("fluently/Fluently-XL-Final", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
+#repo_pixel.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="base")
+#repo_pixel.load_lora_weights("nerijs/pixel-art-xl", adapter_name="base2")
+#repo_pixel.set_adapters(["base", "base2"], adapter_weights=[1, 1])
 
 repo_customs = {
     "Default": repo_default,
-    "Realistic": StableDiffusionXLPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
-    "Anime": StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
-    "Pixel": repo_pixel,
-    "Large": StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16, use_safetensors=True),
+    "Realistic": None,#StableDiffusionXLPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
+    "Anime": None,#StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
+    "Pixel": None,#repo_pixel,
+    "Large": None,#StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16, use_safetensors=True),
 }
 
 # Functions