update app.py

app.py

@@ -167,7 +167,7 @@ with gr.Blocks() as demo:
     session_state = gr.State({})
     default_model_version = "v1.0"
 
-    gr.Markdown("""
+    gr.HTML("""
     <div style="text-align: center; max-width: 900px; margin: 0 auto;">
         <h1 style="font-size: 1.5rem; font-weight: 700; display: block;">InfiniteYou-FLUX</h1>
         <h2 style="font-size: 1.2rem; font-weight: 300; margin-bottom: 1rem; display: block;">Official Gradio Demo for <a href="https://arxiv.org/abs/2503.16418">InfiniteYou: Flexible Photo Recrafting While Preserving Your Identity</a></h2>
@@ -176,7 +176,9 @@ with gr.Blocks() as demo:
         <a href="https://github.com/bytedance/InfiniteYou">[Code]</a>&ensp;
         <a href="https://huggingface.co/ByteDance/InfiniteYou">[Model]</a>
     </div>
+    """)
 
+    gr.Markdown("""
     ### 💡 How to Use This Demo:
     1. **Upload an identity (ID) image containing a human face.** For images with multiple faces, only the largest face will be detected. The face should ideally be clear and large enough, without significant occlusions or blur.
     2. **Enter the text prompt to describe the generated image and select the model version.** Please refer to **important usage tips** under the Generated Image field.
@@ -227,7 +229,7 @@ with gr.Blocks() as demo:
                 - **Model Version**: `aes_stage2` is used by default for better text-image alignment and aesthetics. For higher ID similarity, please try `sim_stage1`.
                 - **Useful Hyperparameters**: Usually, there is NO need to adjust too much. If necessary, try a slightly larger `--infusenet_guidance_start` (*e.g.*, `0.1`) only (especially helpful for `sim_stage1`). If still not satisfactory, then try a slightly smaller `--infusenet_conditioning_scale` (*e.g.*, `0.9`).
                 - **Optional LoRAs**: `realism` and `anti-blur`. To enable them, please check the corresponding boxes. They are optional and were NOT used in our paper.
-                - **Gender Prompt**: If the generated gender is not preferred, add specific words in the text prompt, such as 'a man', 'a woman', *etc*. We encourage using inclusive and respectful language.
+                - **Gender Prompt**: If the generated gender is not preferred, add specific words in the prompt, such as 'a man', 'a woman', *etc*. We encourage using inclusive and respectful language.
                 """
             )
 
@@ -279,7 +281,11 @@ with gr.Blocks() as demo:
         the [FLUX.1-dev](https://huggingface.co/black-forest-labs/FLUX.1-dev) base model, LoRAs, etc., must follow their original licenses and be used only for academic research purposes.
 
         This research aims to positively impact the Generative AI field. Users are granted freedom to create images using this tool, but they must comply with local laws and use it responsibly. The developers do not assume any responsibility for potential misuse.
-        
+        """
+    )    
+
+    gr.Markdown(
+        """
         ### 📖 Citation
 
         If you find InfiniteYou useful for your research or applications, please cite our paper: