test

app.py

@@ -31,15 +31,14 @@ def get_benchmark_df(benchmark):
         llm_perf_dataset_repo.git_pull()
 
     # load
-    df = pd.read_csv(f"./llm-perf-dataset/reports/{benchmark}/inference_report.csv")
+    df = pd.read_csv(
+        f"./llm-perf-dataset/reports/{benchmark}/inference_report.csv")
     # preprocess
     df["model"] = df["model"].apply(make_clickable_model)
     # filter
     df = df[COLUMNS_MAPPING.keys()]
     # rename
-    df.rename(columns={
-        df_col: rename_col for df_col, rename_col in COLUMNS_MAPPING.items()
-    }, inplace=True)
+    df.rename(columns=COLUMNS_MAPPING, inplace=True)
     # sort
     df.sort_values(by=SORTING_COLUMN, ascending=False, inplace=True)
 
@@ -87,13 +86,12 @@ with demo:
             )
 
     with gr.Row():
-        with gr.Column():
-            with gr.Accordion("📙 Citation", open=False):
-                citation_button = gr.Textbox(
-                    value=CITATION_BUTTON_TEXT,
-                    label=CITATION_BUTTON_LABEL,
-                    elem_id="citation-button",
-                ).style(show_copy_button=True)
+        with gr.Accordion("📙 Citation", open=False):
+            citation_button = gr.Textbox(
+                value=CITATION_BUTTON_TEXT,
+                label=CITATION_BUTTON_LABEL,
+                elem_id="citation-button",
+            ).style(show_copy_button=True)
 
 # Restart space every hour
 scheduler = BackgroundScheduler()

src/assets/text_content.py

@@ -18,6 +18,7 @@ CITATION_BUTTON_TEXT = r"""@misc{open-llm-perf-leaderboard,
   howpublished = "\url{https://huggingface.co/spaces/optimum/llm-perf-leaderboard}",
 @software{optimum-benchmark,
   author       = {Ilyas Moutawwakil},
+  publisher = {Hugging Face},
   title        = {A framework for benchmarking the performance of Transformers models on different hardwares and backends},
 }
 """