Merge pull request #6 from AIR-Bench/release-arxiv-paper

app.py

@@ -4,7 +4,10 @@ import gradio as gr
 from apscheduler.schedulers.background import BackgroundScheduler
 from huggingface_hub import snapshot_download
 
-from src.about import BENCHMARKS_TEXT, EVALUATION_QUEUE_TEXT, INTRODUCTION_TEXT, TITLE
+from src.about import (
+    BENCHMARKS_TEXT, EVALUATION_QUEUE_TEXT, INTRODUCTION_TEXT, TITLE,
+    CITATION_BUTTON_LABEL, CITATION_BUTTON_TEXT
+)
 from src.benchmarks import LongDocBenchmarks, QABenchmarks
 from src.columns import COL_NAME_RERANKING_MODEL, COL_NAME_RETRIEVAL_MODEL
 from src.components import (
@@ -615,6 +618,9 @@ with demo:
         with gr.TabItem("📝 About", elem_id="llm-benchmark-tab-table", id=3):
             gr.Markdown(BENCHMARKS_TEXT, elem_classes="markdown-text")
 
+    gr.Markdown(f"{CITATION_BUTTON_LABEL}\n\n{CITATION_BUTTON_TEXT}", elem_classes="markdown-text")
+
+
 if __name__ == "__main__":
     scheduler = BackgroundScheduler()
     scheduler.add_job(restart_space, "interval", seconds=1800)

src/about.py

@@ -32,6 +32,7 @@ EVALUATION_QUEUE_TEXT = """
 
 CITATION_BUTTON_LABEL = "Copy the following snippet to cite these results"
 CITATION_BUTTON_TEXT = r"""
+```bibtex
 @misc{chen2024airbench,
       title={AIR-Bench: Automated Heterogeneous Information Retrieval Benchmark}, 
       author={Jianlyu Chen and Nan Wang and Chaofan Li and Bo Wang and Shitao Xiao and Han Xiao and Hao Liao and Defu Lian and Zheng Liu},
@@ -41,4 +42,5 @@ CITATION_BUTTON_TEXT = r"""
       primaryClass={cs.IR},
       url={https://arxiv.org/abs/2412.13102}, 
 }
+```
 """