Add header and footer

app.py

@@ -71,6 +71,28 @@ def infer(
 
     return images
 
+
+### Description
+header = r"""
+<h1>ID-Booth: Identity-consistent Face Generation with Diffusion Models</h1>
+<b>Official Gradio demo</b> for <a href='https://dariant.github.io/publications/ID-Booth' target='_blank'><b>ID-Booth: Identity-consistent Face Generation with Diffusion Models</b></a>
+"""
+
+footer = r"""
+---
+📝 **Citation**
+<br>
+If you find ID-Booth helpful, please consider citing our paper:
+```bibtex
+@article{tomasevic2025IDBooth,
+  title={{ID-Booth}: Identity-consistent Face Generation with Diffusion Models},
+  author={Toma{\v{s}}evi{\'c}, Darian and Boutros, Fadi and Lin, Chenhao and Damer, Naser and {\v{S}}truc, Vitomir and Peer, Peter},
+  journal={arXiv preprint arXiv:2504.07392},
+  year={2025}
+}
+```
+"""
+
 css = """
 #col-container {
     margin: 0 auto;
@@ -79,6 +101,9 @@ css = """
 """
 
 with gr.Blocks(css=css) as demo:
+
+    # description
+    gr.Markdown(header)
     with gr.Column(elem_id="col-container"):
         gr.Markdown(" # ID-Booth Demo")
 
@@ -183,5 +208,7 @@ with gr.Blocks(css=css) as demo:
         outputs=[result],
     )
 
+    gr.Markdown(footer)
+
 if __name__ == "__main__":
     demo.launch()