Update app.py

app.py

@@ -250,6 +250,7 @@ dtype = torch.float16 if device.type == "cuda" else torch.float32
 # Load primary model (RealVisXL_V5.0_Lightning)
 if torch.cuda.is_available():
     pipe = StableDiffusionXLPipeline.from_pretrained(
+        #"SG161222/RealVisXL_V5.0",
         "SG161222/RealVisXL_V5.0_Lightning",
         torch_dtype=dtype,
         use_safetensors=True,
@@ -270,7 +271,8 @@ if torch.cuda.is_available():
     
     # Load second model (RealVisXL_V4.0)
     pipe2 = StableDiffusionXLPipeline.from_pretrained(
-        "SG161222/RealVisXL_V4.0",
+        #"SG161222/RealVisXL_V4.0",
+        "SG161222/RealVisXL_V4.0_Lightning",
         torch_dtype=dtype,
         use_safetensors=True,
         add_watermarker=False,
@@ -441,7 +443,7 @@ h1 {
 }
 '''
 
-title = """<h1 align="center">IMAGINEO 4K : SDXL 🔥</h1>
+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>
@@ -460,7 +462,7 @@ with gr.Blocks(theme="YTheme/Minecraft", css=css) as demo:
                 placeholder="Enter your prompt",
                 container=False,
             )
-            run_button = gr.Button("Generate Image As ( 1024 x 1024 ) 🧤", scale=0)
+            run_button = gr.Button("Generate Image ( 1024 x 1024 ) 🧤", scale=0)
             
             with gr.Row(visible=True):
                 model_selection = gr.Dropdown(