Update app.py

app.py

@@ -287,9 +287,9 @@ if torch.cuda.is_available():
         pipe2.unet = torch.compile(pipe2.unet, mode="reduce-overhead", fullgraph=True)
         print("Model RealVisXL_V4.0 Compiled!")
     
-    # Load third model (Animagine XL 4.0)
+    # Load third model 
     pipe3 = StableDiffusionXLPipeline.from_pretrained(
-        "cagliostrolab/animagine-xl-4.0",
+        "SG161222/RealVisXL_V3.0_Turbo",
         torch_dtype=dtype,
         use_safetensors=True,
         add_watermarker=False,
@@ -320,7 +320,7 @@ else:
         add_watermarker=False,
     ).to(device)
     pipe3 = StableDiffusionXLPipeline.from_pretrained(
-        "cagliostrolab/animagine-xl-4.0",
+        "SG161222/RealVisXL_V3.0_Turbo",
         torch_dtype=dtype,
         use_safetensors=True,
         add_watermarker=False,
@@ -329,11 +329,11 @@ else:
 
 # A dictionary to easily choose the model based on selection.
 DEFAULT_MODEL = "Lightning 5"
-MODEL_CHOICES = [DEFAULT_MODEL, "Lightning 4", "Animagine 4"]
+MODEL_CHOICES = [DEFAULT_MODEL, "Lightning 4", "Turbo v3"]
 models = {
     "Lightning 5": pipe,
     "Lightning 4": pipe2,
-    "Animagine 4": pipe3
+    "Turbo v3": pipe3
 }
 
 def save_image(img, path):
@@ -452,7 +452,7 @@ title = """<h1 align="center">IMAGINEO-4K : SDXL</h1>
 <p><center>
 <a href="https://huggingface.co/SG161222/RealVisXL_V4.0_Lightning" target="_blank">[Lightning 4]</a>
 <a href="https://huggingface.co/SG161222/RealVisXL_V5.0_Lightning" target="_blank">[Lightning 5]</a>
-<a href="https://huggingface.co/cagliostrolab/animagine-xl-4.0" target="_blank">[Animagine 4]</a>
+<a href="https://huggingface.co/SG161222/RealVisXL_V3.0_Turbo" target="_blank">[Turbo v3]</a>
 </center></p>
 """
 
@@ -467,7 +467,7 @@ with gr.Blocks(css=css) as demo:
                 placeholder="Enter your prompt :/",
                 container=False,
             )
-            run_button = gr.Button("Generate Image as ( 1024 x 1024 ) 🧤", scale=0, elem_classes="submit-btn")
+            run_button = gr.Button("Generate Image as ( 1024 x 1024 ) 🔥", scale=0, elem_classes="submit-btn")
             
             with gr.Row(visible=True):
                 model_selection = gr.Dropdown(