Add citation information (#32)

README.md

@@ -60,3 +60,25 @@ You can still run benchmarks without Pegasus like this:
 $ docker exec leaderboard0 python scripts/benchmark.py --model-path /data/leaderboard/weights/lmsys/vicuna-13B --input-file sharegpt/sg_90k_part1_html_cleaned_lang_first_sampled_sorted.json
 $ docker exec leaderboard0 python scripts/benchmark.py --model-path databricks/dolly-v2-12b --input-file sharegpt/sg_90k_part1_html_cleaned_lang_first_sampled_sorted.json
 ```
+
+## Citation
+
+For the ML.ENERGY Leaderboard:
+```bibtex
+@misc{ml-energy-leaderboard,
+    author={Jae-Won Chung and Jiachen Liu and Zhiyu Wu and Yuxuan Xia and Mosharaf Chowdhury},
+    title={{ML.ENERGY} Leaderboard},
+    year={2023},
+    howpublished={\url{https://ml.energy/leaderboard}},
+}
+```
+
+For the [Zeus](https://ml.energy/zeus) framework used for benchmarking:
+```bibtex
+@inproceedings{zeus:nsdi23,
+    author={Jie You and Jae-Won Chung and Mosharaf Chowdhury},
+    booktitle={USENIX NSDI},
+    title={Zeus: Understanding and Optimizing {GPU} Energy Consumption of {DNN} Training},
+    year={2023},
+}
+```

app.py

@@ -357,6 +357,11 @@ table th:first-child {
   pointer-events: none;
   opacity: 0.4;
 }
+
+/* Make the Citation header larger */
+#citation-header > div > span {
+    font-size: 16px !important;
+}
 """
 
 intro_text = """
@@ -790,6 +795,17 @@ with gr.Blocks(css=custom_css) as block:
             # Read in LEADERBOARD.md
             gr.Markdown(open("docs/leaderboard.md").read())
 
+    # Citation
+    with gr.Accordion("📚  Citation", open=False, elem_id="citation-header"):
+        citation_text = open("docs/citation.bib").read()
+        gr.Textbox(
+            value=citation_text,
+            label="BibTeX for the leaderboard and the Zeus framework used for benchmarking:",
+            lines=len(list(filter(lambda c: c == "\n", citation_text))),
+            interactive=False,
+            show_copy_button=True,
+        )
+
     # Load the table on page load.
     block.load(on_load, outputs=[dataframe, model_preference_dropdown], queue=False)
 

docs/citation.bib

@@ -0,0 +1,12 @@
+@misc{ml-energy-leaderboard,
+    author={Jae-Won Chung and Jiachen Liu and Zhiyu Wu and Yuxuan Xia and Mosharaf Chowdhury},
+    title={{ML.ENERGY} Leaderboard},
+    year={2023},
+    howpublished={\url{https://ml.energy/leaderboard}},
+}
+@inproceedings{zeus:nsdi23,
+    author={Jie You and Jae-Won Chung and Mosharaf Chowdhury},
+    booktitle={USENIX NSDI},
+    title={Zeus: Understanding and Optimizing {GPU} Energy Consumption of {DNN} Training},
+    year={2023},
+}