Update app.py

app.py

@@ -11,7 +11,7 @@ import os
 import numpy as np
 
 from huggingface_hub import hf_hub_download
-from diffusers import DiffusionPipeline,StableDiffusionPipeline, DPMSolverMultistepScheduler
+from diffusers import DiffusionPipeline, StableDiffusionPipeline, DPMSolverMultistepScheduler
 from transformers import pipeline
 from PIL import Image
 
@@ -43,20 +43,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 = StableDiffusionPipeline.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_pro = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, use_safetensors=True)
+repo_pro = StableDiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, use_safetensors=True)
 repo_pro.load_lora_weights(hf_hub_download("alimama-creative/FLUX.1-Turbo-Alpha", "diffusion_pytorch_model.safetensors"))
 
 repo_classic = StableDiffusionPipeline.from_pretrained("gsdf/Counterfeit-V2.5", torch_dtype=torch.float16, use_safetensors=True, safety_checker=None)
 
 repo_customs = {
     "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),
+    "Realistic": StableDiffusionPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
+    "Anime": StableDiffusionPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
     "Pixel": repo_default,
     "Pro": repo_pro,
     "Classic": repo_classic,