Update app.py

app.py

@@ -63,9 +63,9 @@ 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_pony = snapshot_download(repo_id="John6666/pony-realism-v22main-sdxl")
+ckpt_dir_cyber = snapshot_download(repo_id="John6666/cyberrealistic-pony-v65-sdxl")
+ckpt_dir_stallion = snapshot_download(repo_id="TheImposterImposters/tamePony_v25")
 
 # Load the models
 vae_pony = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir_pony, "vae"), torch_dtype=torch.float16)
@@ -211,11 +211,11 @@ 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":
+    if model_choice == "Pony Realism v22":
         pipe = pipe_pony
-    elif model_choice == "Cyber Realistic Pony v63":
+    elif model_choice == "Cyber Realistic Pony v65":
         pipe = pipe_cyber
-    else:  # "Stallion Dreams Pony Realistic v1"
+    else:  # "Tame Pony"
         pipe = pipe_stallion
     
     if use_random_seed:
@@ -300,12 +300,12 @@ with gr.Blocks(theme='bethecloud/storj_theme') as demo:
     gr.HTML("""
     <h1 align="center">Pony Realism / Cyber Realism / Stallion Dreams</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://huggingface.co/John6666/pony-realism-v22main-sdxl/" target="_blank">[Pony Realism]</a>
+    <a href="https://huggingface.co/John6666/cyberrealistic-pony-v65-sdxl" target="_blank">[Cyberrealistic Pony]</a>
+    <a href="https://huggingface.co/TheImposterImposters/tamePony_v25" target="_blank">[Tame Pony]</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://civitai.com/models/722045/tame-pony-the-authenticity-machine" target="_blank">[Tame Pony civitai]</a>
     <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 +315,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"], 
+                    ["Pony Realism v22", "Cyber Realistic Pony v65", "Tame Pony v25"], 
                     label="Model Choice", 
-                    value="Pony Realism v21")
+                    value="Pony Realism v22")
             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")