Update app.py

app.py

@@ -41,21 +41,20 @@ footer {
 
 repo_nsfw_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection")
 
-repo_default = DiffusionPipeline.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="default_base")
-repo_default.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="pixel_base")
-repo_default.load_lora_weights("nerijs/pixel-art-xl", adapter_name="pixel_base_2")
+# repo_default = DiffusionPipeline.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="default_base")
+# repo_default.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="pixel_base")
+# repo_default.load_lora_weights("nerijs/pixel-art-xl", adapter_name="pixel_base_2")
 
-# repo_large = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, add_watermarker=False, revision="refs/pr/1")
+repo_large = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
 # repo_large.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
-# repo_large.set_adapters(["base"], adapter_weights=[0.7])
 
 repo_customs = {
-    "Default": repo_default,
+    # "Default": repo_default,
     # "Realistic": DiffusionPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
     # "Anime": DiffusionPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
-    "Pixel": repo_default,
-    # "Large": repo_neo,
+    # "Pixel": repo_default,
+    "Large": repo_large,
 }
 
 # Functions