Update app.py

app.py

@@ -63,60 +63,60 @@ download_file("https://huggingface.co/ai-forever/Real-ESRGAN/resolve/main/RealES
 download_file("https://huggingface.co/ai-forever/Real-ESRGAN/resolve/main/RealESRGAN_x4.pth?download=true", "models/upscalers/", "RealESRGAN_x4.pth")
 
 # Download the model files
-ckpt_dir_pony = snapshot_download(repo_id="John6666/pony-realism-v21main-sdxl")
-ckpt_dir_cyber = snapshot_download(repo_id="John6666/cyberrealistic-pony-v63-sdxl")
-ckpt_dir_stallion = snapshot_download(repo_id="John6666/stallion-dreams-pony-realistic-v1-sdxl")
+ckpt_dir_realpony = snapshot_download(repo_id="silveroxides/RNS_RealPonyV20")
+ckpt_dir_ultpony = snapshot_download(repo_id="silveroxides/RNS_PonyUltimateV20")
+ckpt_dir_hybridpony = snapshot_download(repo_id="silveroxides/RealHybridPony")
 
 # Load the models
-vae_pony = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir_pony, "vae"), torch_dtype=torch.float16)
-vae_cyber = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir_cyber, "vae"), torch_dtype=torch.float16)
-vae_stallion = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir_stallion, "vae"), torch_dtype=torch.float16)
+vae_realpony = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir_realpony, "vae"), torch_dtype=torch.float16)
+vae_ultpony = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir_ultpony, "vae"), torch_dtype=torch.float16)
+vae_hybridpony = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir_hybridpony, "vae"), torch_dtype=torch.float16)
 
-pipe_pony = StableDiffusionXLPipeline.from_pretrained(
-    ckpt_dir_pony,
-    vae=vae_pony,
+pipe_realpony = StableDiffusionXLPipeline.from_pretrained(
+    ckpt_dir_realpony,
+    vae=vae_realpony,
     torch_dtype=torch.float16,
     use_safetensors=True,
 )
-pipe_cyber = StableDiffusionXLPipeline.from_pretrained(
-    ckpt_dir_cyber,
-    vae=vae_cyber,
+pipe_ultpony = StableDiffusionXLPipeline.from_pretrained(
+    ckpt_dir_ultpony,
+    vae=vae_ultpony,
     torch_dtype=torch.float16,
     use_safetensors=True,
 )
-pipe_stallion = StableDiffusionXLPipeline.from_pretrained(
-    ckpt_dir_stallion,
-    vae=vae_stallion,
+pipe_hybridpony = StableDiffusionXLPipeline.from_pretrained(
+    ckpt_dir_hybridpony,
+    vae=vae_hybridpony,
     torch_dtype=torch.float16,
     use_safetensors=True,
 )
 
-pipe_pony = pipe_pony.to("cuda")
-pipe_cyber = pipe_cyber.to("cuda")
-pipe_stallion = pipe_stallion.to("cuda")
+pipe_realpony = pipe_realpony.to("cuda")
+pipe_ultpony = pipe_ultpony.to("cuda")
+pipe_hybridpony = pipe_hybridpony.to("cuda")
 
-pipe_pony.unet.set_attn_processor(AttnProcessor2_0())
-pipe_cyber.unet.set_attn_processor(AttnProcessor2_0())
-pipe_stallion.unet.set_attn_processor(AttnProcessor2_0())
+pipe_realpony.unet.set_attn_processor(AttnProcessor2_0())
+pipe_ultpony.unet.set_attn_processor(AttnProcessor2_0())
+pipe_hybridpony.unet.set_attn_processor(AttnProcessor2_0())
 
 # Define samplers
 samplers = {
-    "Euler a": EulerAncestralDiscreteScheduler.from_config(pipe_pony.scheduler.config),
-    "DPM++ SDE Karras": DPMSolverSDEScheduler.from_config(pipe_pony.scheduler.config, use_karras_sigmas=True),
-    "Heun": HeunDiscreteScheduler.from_config(pipe_pony.scheduler.config),
+    "Euler a": EulerAncestralDiscreteScheduler.from_config(pipe_realpony.scheduler.config),
+    "DPM++ SDE Karras": DPMSolverSDEScheduler.from_config(pipe_realpony.scheduler.config, use_karras_sigmas=True),
+    "Heun": HeunDiscreteScheduler.from_config(pipe_realpony.scheduler.config),
     # New samplers
-    "DPM++ 2M SDE Karras": DPMSolverMultistepScheduler.from_config(pipe_pony.scheduler.config, use_karras_sigmas=True, algorithm_type="sde-dpmsolver++"),
-    "DPM++ 2M": DPMSolverMultistepScheduler.from_config(pipe_pony.scheduler.config),
-    "DDIM": DDIMScheduler.from_config(pipe_pony.scheduler.config),
-    "LMS": LMSDiscreteScheduler.from_config(pipe_pony.scheduler.config),
-    "PNDM": PNDMScheduler.from_config(pipe_pony.scheduler.config),
-    "UniPC": UniPCMultistepScheduler.from_config(pipe_pony.scheduler.config),
+    "DPM++ 2M SDE Karras": DPMSolverMultistepScheduler.from_config(pipe_realpony.scheduler.config, use_karras_sigmas=True, algorithm_type="sde-dpmsolver++"),
+    "DPM++ 2M": DPMSolverMultistepScheduler.from_config(pipe_realpony.scheduler.config),
+    "DDIM": DDIMScheduler.from_config(pipe_realpony.scheduler.config),
+    "LMS": LMSDiscreteScheduler.from_config(pipe_realpony.scheduler.config),
+    "PNDM": PNDMScheduler.from_config(pipe_realpony.scheduler.config),
+    "UniPC": UniPCMultistepScheduler.from_config(pipe_realpony.scheduler.config),
 }
 
-DEFAULT_POSITIVE_PREFIX = "Score_9 score_8_up score_7_up BREAK"
-DEFAULT_POSITIVE_SUFFIX = "(masterpiece) very_aesthetic detailed_face cinematic footage"
-DEFAULT_NEGATIVE_PREFIX = "Score_1 score_2 score _3 text low_res"
-DEFAULT_NEGATIVE_SUFFIX = "Nsfw oversaturated crappy_art low_quality blurry bad_anatomy extra_digits fewer_digits simple_background very_displeasing watermark signature"
+DEFAULT_POSITIVE_PREFIX = "score_8_up, score_7_up, accurate, genuine"
+DEFAULT_POSITIVE_SUFFIX = "perfect composition, detailed, photorealism, real life, raw, reality, cinematic"
+DEFAULT_NEGATIVE_PREFIX = "score_1, score_2, score_3, text, artist name, signature, watermark, logo, url, web address"
+DEFAULT_NEGATIVE_SUFFIX = "low quality, low resolution, simple background, bad composition, deformed, disfigured, sketch, unfinished"
 
 # Initialize Florence model
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -211,12 +211,12 @@ def generate_image(model_choice, additional_positive_prompt, additional_negative
                    input_image=None, progress=gr.Progress(track_tqdm=True)):
     
     # Select the appropriate pipe based on the model choice
-    if model_choice == "Pony Realism v21":
-        pipe = pipe_pony
-    elif model_choice == "Cyber Realistic Pony v63":
-        pipe = pipe_cyber
-    else:  # "Stallion Dreams Pony Realistic v1"
-        pipe = pipe_stallion
+    if model_choice == "Real Pony RNS":
+        pipe = pipe_realpony
+    elif model_choice == "Ultimate Pony RNS":
+        pipe = pipe_ultpony
+    else:  # "Hybrid Pony SDXL"
+        pipe = pipe_hybridpony
     
     if use_random_seed:
         seed = random.randint(0, 2**32 - 1)
@@ -298,14 +298,11 @@ def generate_image(model_choice, additional_positive_prompt, additional_negative
 # Gradio interface
 with gr.Blocks(theme='bethecloud/storj_theme') as demo:
     gr.HTML("""
-    <h1 align="center">Pony Realism / Cyber Realism / Stallion Dreams</h1>
+    <h1 align="center">Real Pony RNS / Ultimate Pony RNS / Hybrid Pony SDXL</h1>
     <p align="center">
-    <a href="https://huggingface.co/John6666/pony-realism-v21main-sdxl/" target="_blank">[Pony Realism]</a>
-    <a href="https://huggingface.co/John6666/cyberrealistic-pony-v63-sdxl" target="_blank">[Cyberrealistic Pony]</a>
-    <a href="https://huggingface.co/John6666/stallion-dreams-pony-realistic-v1-sdxl" target="_blank">[Stallion Dreams]</a><br>
-    <a href="https://civitai.com/models/372465/pony-realism" target="_blank">[Pony Realism civitai]</a>
-    <a href="https://civitai.com/models/443821?modelVersionId=680915" target="_blank">[Cyberrealistic Pony civitai]</a>
-    <a href="https://civitai.com/models/628846/stallion-dreams-pony-realistic" target="_blank">[Stallion Dreams civitai]</a>
+    <a href="https://huggingface.co/silveroxides/RNS_RealPonyV20" target="_blank">[Pony Realism]</a>
+    <a href="https://huggingface.co/silveroxides/RNS_PonyUltimateV20" target="_blank">[Cyberrealistic Pony]</a>
+    <a href="https://huggingface.co/silveroxides/RealHybridPony" target="_blank">[Stallion Dreams]</a><br>
     <a href="https://huggingface.co/microsoft/Florence-2-base" target="_blank">[Florence-2 Model]</a>
     <a href="https://huggingface.co/gokaygokay/Lamini-Prompt-Enchance-Long" target="_blank">[Prompt Enhancer Long]</a>
     <a href="https://huggingface.co/gokaygokay/Lamini-Prompt-Enchance" target="_blank">[Prompt Enhancer Medium]</a>
@@ -315,9 +312,9 @@ with gr.Blocks(theme='bethecloud/storj_theme') as demo:
     with gr.Row():
         with gr.Column(scale=1):
             model_choice = gr.Dropdown(
-                    ["Pony Realism v21", "Cyber Realistic Pony v63", "Stallion Dreams Pony Realistic v1"], 
+                    ["Real Pony RNS", "Ultimate Pony RNS", "Hybrid Pony SDXL"], 
                     label="Model Choice", 
-                    value="Pony Realism v21")
+                    value="Real Pony RNS")
             positive_prompt = gr.Textbox(label="Positive Prompt", placeholder="Add your positive prompt here")
             negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="Add your negative prompt here")