added dataframe text

app.py

@@ -26,7 +26,7 @@ SORTING_COLUMN = ["Throughput (tokens/s) ⬆️"]
 llm_perf_dataset_repo = load_dataset_repo(LLM_PERF_DATASET_REPO, OPTIMUM_TOKEN)
 
 
-def get_vanilla_benchmark_df():
+def get_benchmark_df():
     if llm_perf_dataset_repo:
         llm_perf_dataset_repo.git_pull()
 
@@ -58,17 +58,22 @@ with demo:
     gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
 
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
-        with gr.TabItem("A100 Benchmark", elem_id="vanilla-benchmark", id=0):
-            vanilla_benchmark_df = get_vanilla_benchmark_df()
-
-            A100_text = "Machine: 4x A100 80GB<br>Batches: 1<br>Number of tokens: 100<br>"
+        with gr.TabItem("A100 Benchmark", elem_id="a100-benchmark", id=0):
+            A100_text = "<h1>Machine: 4x A100 80GB<h1>"
             gr.HTML(A100_text)
 
+            dataframe_text = """
+            <h3>Batch Size: 1</h3>
+            <h3>Generated Tokens: 100</h3>
+            """
+
+            gr.HTML(dataframe_text)
+            benchmark_df = get_benchmark_df()
             leaderboard_table_lite = gr.components.Dataframe(
-                value=vanilla_benchmark_df,
+                value=benchmark_df,
                 datatype=COLUMNS_DATATYPES,
                 headers=NEW_COLUMNS,
-                elem_id="vanilla-benchmark",
+                elem_id="pytorch-a100-benchmark",
             )