Update app.py

app.py

@@ -64,12 +64,12 @@ download_file("https://huggingface.co/ai-forever/Real-ESRGAN/resolve/main/RealES
 
 # Download the model files
 ckpt_dir_realpony = snapshot_download(repo_id="silveroxides/RNS_RealPonyV20")
-ckpt_dir_ultpony = snapshot_download(repo_id="silveroxides/RNS_PonyUltimateV20")
+ckpt_dir_noobai = snapshot_download(repo_id="silveroxides/NoobAI-XL-EPS-1.0-Vwe")
 ckpt_dir_hybridpony = snapshot_download(repo_id="silveroxides/RealHybridPony")
 
 # Load the models
 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_noobai = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir_noobai, "vae"), torch_dtype=torch.float16)
 vae_hybridpony = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir_hybridpony, "vae"), torch_dtype=torch.float16)
 
 pipe_realpony = StableDiffusionXLPipeline.from_pretrained(
@@ -79,9 +79,9 @@ pipe_realpony = StableDiffusionXLPipeline.from_pretrained(
     torch_dtype=torch.float16,
     use_safetensors=True,
 )
-pipe_ultpony = StableDiffusionXLPipeline.from_pretrained(
-    ckpt_dir_ultpony,
-    vae=vae_ultpony,
+pipe_noobai = StableDiffusionXLPipeline.from_pretrained(
+    ckpt_dir_noobai,
+    vae=vae_noobai,
     custom_pipeline="multimodalart/sdxl_perturbed_attention_guidance",
     torch_dtype=torch.float16,
     use_safetensors=True,
@@ -95,11 +95,11 @@ pipe_hybridpony = StableDiffusionXLPipeline.from_pretrained(
 )
 
 pipe_realpony = pipe_realpony.to("cuda")
-pipe_ultpony = pipe_ultpony.to("cuda")
+pipe_noobai = pipe_noobai.to("cuda")
 pipe_hybridpony = pipe_hybridpony.to("cuda")
 
 pipe_realpony.unet.set_attn_processor(AttnProcessor2_0())
-pipe_ultpony.unet.set_attn_processor(AttnProcessor2_0())
+pipe_noobai.unet.set_attn_processor(AttnProcessor2_0())
 pipe_hybridpony.unet.set_attn_processor(AttnProcessor2_0())
 
 # Define samplers
@@ -216,8 +216,8 @@ def generate_image(model_choice, additional_positive_prompt, additional_negative
     # Select the appropriate pipe based on the model choice
     if model_choice == "Real Pony RNS":
         pipe = pipe_realpony
-    elif model_choice == "Ultimate Pony RNS":
-        pipe = pipe_ultpony
+    elif model_choice == "NoobAI-XL EPS 1.0":
+        pipe = pipe_noobai
     else:  # "Hybrid Pony SDXL"
         pipe = pipe_hybridpony
     
@@ -305,9 +305,9 @@ with gr.Blocks(theme='bethecloud/storj_theme') as demo:
     gr.HTML("""
     <h1 align="center">Real Pony RNS / Ultimate Pony RNS / Hybrid Pony SDXL</h1>
     <p align="center">
-    <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/silveroxides/RNS_RealPonyV20" target="_blank">[RNS RealPony]</a>
+    <a href="https://huggingface.co/silveroxides/NoobAI-XL-EPS-1.0-Vwe" target="_blank">[NoobAI-XL EPS 1.0]</a>
+    <a href="https://huggingface.co/silveroxides/RealHybridPony" target="_blank">[Hybrid Pony SDXL]</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>
@@ -317,7 +317,7 @@ with gr.Blocks(theme='bethecloud/storj_theme') as demo:
     with gr.Row():
         with gr.Column(scale=1):
             model_choice = gr.Dropdown(
-                    ["Real Pony RNS", "Ultimate Pony RNS", "Hybrid Pony SDXL"], 
+                    ["Real Pony RNS", "NoobAI-XL EPS 1.0", "Hybrid Pony SDXL"], 
                     label="Model Choice", 
                     value="Real Pony RNS")
             positive_prompt = gr.Textbox(label="Positive Prompt", placeholder="Add your positive prompt here")