Update app.py

app.py

@@ -314,7 +314,7 @@ else:
         add_watermarker=False
     ).to(device)
     pipe2 = StableDiffusionXLPipeline.from_pretrained(
-        "SG161222/RealVisXL_V4.0",
+        "SG161222/RealVisXL_V4.0_Lightning",
         torch_dtype=dtype,
         use_safetensors=True,
         add_watermarker=False,
@@ -447,7 +447,16 @@ h1 {
 }
 '''
 
+title = """<h1 align="center">FLUX.1-dev with Qwen2VL Captioner and Prompt Enhancer</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_V4.0_Lightning" target="_blank">[Lightning 4]</a>
+<a href="https://huggingface.co/cagliostrolab/animagine-xl-4.0" target="_blank">[Animagine 4]</a>
+</center></p>
+"""
+
 with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
+    gr.HTML(title)
     with gr.Row():
         with gr.Column(scale=1):
             prompt = gr.Text(