2024-02-19 16:07:07 Publish script update

app.py

@@ -118,7 +118,7 @@ def dataframes():
 
     def for_dataframe(df):
         if not searched_model_names:
-            return df
+            return dataframe_style(df)
         return dataframe_style(pd.concat(df[df.model.str.lower().str.contains(n)] for n in searched_model_names))
 
     return (
@@ -130,14 +130,23 @@ def dataframes():
 
 def dataframe_style(df: pd.DataFrame):
     df = df.copy()
-    df.columns = [snake_case_to_title(column) for column in df.columns]
     column_formats = {}
+    new_column_names = []
+
     for column in df.columns:
         if is_numeric_dtype(df[column]):
             if column == "execution_time":
                 column_formats[column] = "{:.4f}"
             else:
                 column_formats[column] = "{:.2f}"
+        new_column_name = snake_case_to_title(column)
+        if "time" in column:
+            new_column_name += " (Seconds)"
+        elif "chunk" in column:
+            new_column_name += " (Characters)"
+        new_column_names.append(new_column_name)
+    
+    df.columns = new_column_names
     df = df.style.format(column_formats, na_rep="")
     return df
 
@@ -196,7 +205,7 @@ def display_plot(plot_df_row):
         gr.Markdown(str(row["description"]))
 
 
-with gr.Blocks() as demo:
+with gr.Blocks(theme=gr.themes.Default(text_size="lg")) as demo:
     gr.HTML("<h1>Performance LLM Board</h1>")
 
     with gr.Row():