update

pages/arena.py

@@ -6,6 +6,7 @@ import json
 from utils.data import dataset
 from utils.multiple_stream import stream_data
 from pages.summarization_playground import get_model_batch_generation
+from pages.summarization_playground import custom_css
 
 def random_data_selection():
     datapoint = random.choice(dataset)
@@ -18,7 +19,7 @@ def create_arena():
         json_data = file.read()
         prompts = json.loads(json_data)
 
-    with gr.Blocks() as demo:
+    with gr.Blocks(theme=gr.themes.Soft(spacing_size="sm",text_size="sm"), css=custom_css) as demo:
         with gr.Group():
             datapoint = random_data_selection()
             gr.Markdown("""This arena is designed to compare different prompts. Click the button to stream responses from randomly shuffled prompts. Each column represents a response generated from one randomly selected prompt.

pages/batch_evaluation.py

@@ -11,6 +11,7 @@ from utils.model import Model
 from utils.metric import metric_rouge_score
 
 from pages.summarization_playground import generate_answer
+from pages.summarization_playground import custom_css
 
 load_dotenv()
 
@@ -83,7 +84,7 @@ def process(model_selection, prompt, num=10):
 
 
 def create_batch_evaluation_interface():
-    with gr.Blocks() as demo:
+    with gr.Blocks(theme=gr.themes.Soft(spacing_size="sm",text_size="sm"), css=custom_css) as demo:
         gr.Markdown("## Here are evaluation setups. It will run though datapoints in test_data.josn to generate and evaluate. Show results once finished.")
 
         model_dropdown = gr.Dropdown(choices=Model.__model_list__, label="Choose a model", value=Model.__model_list__[0])

pages/leaderboard.py

@@ -1,6 +1,8 @@
 import gradio as gr
 import pandas as pd
 
+from pages.summarization_playground import custom_css
+
 # Sample data for the leaderboard
 data = {
     'Rank': [1, 2, 3, 4, 5],
@@ -31,7 +33,7 @@ def update_leaderboard(sort_by):
     return html
 
 def create_leaderboard():
-    with gr.Blocks(css="#leaderboard table { width: 100%; } #leaderboard th, #leaderboard td { padding: 8px; text-align: left; }") as demo:
+    with gr.Blocks(theme=gr.themes.Soft(spacing_size="sm",text_size="sm"), css=custom_css) as demo:
         gr.Markdown("# 🏆 Summarization Arena Leaderboard")
 
         with gr.Row():