table test

app.py

@@ -3,6 +3,7 @@ import argparse
 import os
 import boto3
 from datetime import datetime
+import pandas as pd
 
 from utils import generate, send_to_s3
 from models import get_tiny_llama, response_tiny_llama
@@ -32,6 +33,27 @@ def tab_arena():
         
     # arena.launch()
 
+def tab_leaderboard():
+    df = pd.DataFrame({
+        "Model" : ['A', 'B', 'C',], 
+        "Test 1" : [0, 1, 0], 
+        "Test 2" : [1, 0, 1,],
+    })
+
+    # Function to apply text color
+    def highlight_cols(x): 
+        df = x.copy() 
+        # df.loc[:, :] = 'color: purple'
+        df[['Model']] = 'color: green'
+        return df 
+
+    # Applying the style function
+    # s = df.style.apply(highlight_cols, axis = None)
+
+    # Displaying the styled dataframe in Gradio
+    with gradio.Blocks() as demo:
+        gradio.DataFrame(df)
+
 with open("test.md", "r") as f:
     TEST_MD = f.read()
 
@@ -47,7 +69,7 @@ def build_demo():
         with gradio.Tabs(elem_classes="tab-buttons") as tabs:
             with gradio.TabItem("🐼 MERA leaderboard", elem_id="od-benchmark-tab-table", id=0):
                 gradio.Markdown(TEST_MD, elem_classes="markdown-text-details")
-                # _tab_leaderboard()
+                tab_leaderboard()
 
             with gradio.TabItem("🆚 SBS by categories and criteria", elem_id="od-benchmark-tab-table", id=1):
                 gradio.Markdown(TEST_MD, elem_classes="markdown-text-details")