Update app.py

app.py

@@ -4,7 +4,7 @@ __all__ = ['block', 'make_clickable_model', 'make_clickable_user', 'get_submissi
 import gradio as gr
 import pandas as pd
 
-COLUMN_NAMES = ["model", "Tuned on ToolBench", "Open Weather", "The Cat API", "Home Search", "Trip Booking", "Google Sheets", "VirtualHome", "WebShop Long", "WebShop Short", "Tabletop"]
+COLUMN_NAMES = ["model", "Tuned on ToolBench", "Avg.", "Open Weather", "The Cat API", "Home Search", "Trip Booking", "Google Sheets", "VirtualHome", "WebShop Long", "WebShop Short", "Tabletop"]
 UNTUNED_MODEL_RESULTS = '''[gpt4](https://platform.openai.com/docs/models/gpt-4)                    & 93.0 & 96.0 & 97.0 & 96.7 & 62.9 & 23.0 / 23.5 & 0.0 & 0.0 & 81.0 \\
 [text-davinci-003](https://platform.openai.com/docs/models/gpt-3)      & 99.0 & 98.0 & 97.0 & 89.2 & 62.9 & 31.0 / 25.1 & 0.0 & 0.0 & 66.7 \\
 [gpt-3.5-turbo](https://platform.openai.com/docs/models/gpt-3-5)           & 90.0 & 92.0 & 80.0 & 85.8 & 51.4 & 20.0 / 18.9 & 0.0        & 1.8        & 33.3 \\
@@ -56,12 +56,16 @@ def get_baseline_df():
     for line in lines:
         model_results = parse_line(line)
         assert len(model_results) == 10
+        avg = sum(model_results[1:]) / 9
+        model_results.insert(1, avg)
         model_results.insert(1, "False")
         df_data.append(model_results)
     lines = TUNED_MODEL_RESULTS.split("\n")
     for line in lines:
         model_results = parse_line(line)
         assert len(model_results) == 10
+        avg = sum(model_results[1:]) / 9
+        model_results.insert(1, avg)
         model_results.insert(1, "True")
         df_data.append(model_results)
         
@@ -110,7 +114,7 @@ with block:
     )
     with gr.Row():
         data = gr.components.Dataframe(
-            type="pandas", datatype=["markdown", "markdown", "number", "number", "number", "number", "number", "number", "number", "number", "number"]
+            type="pandas", datatype=["markdown", "markdown", "number", "number", "number", "number", "number", "number", "number", "number", "number", "number"]
         )
     with gr.Row():
         data_run = gr.Button("Refresh")