bug fix

app.py

@@ -10,7 +10,7 @@ from src.utils import restart_space, load_dataset_repo, make_clickable_model
 
 LLM_PERF_LEADERBOARD_REPO = "optimum/llm-perf-leaderboard"
 LLM_PERF_DATASET_REPO = "optimum/llm-perf-dataset"
-OPTIMUM_TOKEN = os.environ.get("OPTIMUM_TOKEN")
+OPTIMUM_TOKEN = os.environ.get("OPTIMUM_TOKEN", None)
 
 COLUMNS_MAPPING = {
     "model": "Model 🤗",
@@ -36,7 +36,7 @@ def get_benchmark_df(benchmark):
     # preprocess
     df["model"] = df["model"].apply(make_clickable_model)
     # filter
-    df = df[COLUMNS_MAPPING.keys()]
+    df = df[list(COLUMNS_MAPPING.keys())]
     # rename
     df.rename(columns=COLUMNS_MAPPING, inplace=True)
     # sort
@@ -52,41 +52,39 @@ with demo:
     gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
 
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
-        with gr.Row():
-            with gr.TabItem("🖥️ A100-80GB Benchmark 🏋️", elem_id="A100-benchmark", id=0):
-
-                SINGLE_A100_TEXT = """<h3>Single-GPU (1xA100):</h3>
-                <ul>
-                    <li>Singleton Batch (1)</li>
-                    <li>Thousand Tokens (1000)</li>
-                </ul>
-                """
-                gr.HTML(SINGLE_A100_TEXT)
-
-                single_A100_df = get_benchmark_df(benchmark="1xA100-80GB")
-                leaderboard_table_lite = gr.components.Dataframe(
-                    value=single_A100_df,
-                    datatype=COLUMNS_DATATYPES,
-                    headers=COLUMNS_MAPPING.values(),
-                    elem_id="1xA100-table",
-                )
-        
-        with gr.Row():
-            MULTI_A100_TEXT = """<h3>Multi-GPU (4xA100):</h3>
+        with gr.TabItem("🖥️ A100-80GB Benchmark 🏋️", elem_id="A100-benchmark", id=0):
+
+            SINGLE_A100_TEXT = """<h3>Single-GPU (1xA100):</h3>
             <ul>
                 <li>Singleton Batch (1)</li>
                 <li>Thousand Tokens (1000)</li>
-            </ul>"""
-            gr.HTML(MULTI_A100_TEXT)
+            </ul>
+            """
+            gr.HTML(SINGLE_A100_TEXT)
 
-            multi_A100_df = get_benchmark_df(benchmark="4xA100-80GB")
-            leaderboard_table_full = gr.components.Dataframe(
-                value=multi_A100_df,
+            single_A100_df = get_benchmark_df(benchmark="1xA100-80GB")
+            leaderboard_table_lite = gr.components.Dataframe(
+                value=single_A100_df,
                 datatype=COLUMNS_DATATYPES,
-                headers=COLUMNS_MAPPING.values(),
-                elem_id="4xA100-table",
+                headers=list(COLUMNS_MAPPING.values()),
+                elem_id="1xA100-table",
             )
 
+        MULTI_A100_TEXT = """<h3>Multi-GPU (4xA100):</h3>
+        <ul>
+            <li>Singleton Batch (1)</li>
+            <li>Thousand Tokens (1000)</li>
+        </ul>"""
+        gr.HTML(MULTI_A100_TEXT)
+
+        multi_A100_df = get_benchmark_df(benchmark="4xA100-80GB")
+        leaderboard_table_full = gr.components.Dataframe(
+            value=multi_A100_df,
+            datatype=COLUMNS_DATATYPES,
+            headers=list(COLUMNS_MAPPING.values()),
+            elem_id="4xA100-table",
+        )
+
     with gr.Row():
         with gr.Accordion("📙 Citation", open=False):
             citation_button = gr.Textbox(