metrics

app.py

@@ -51,8 +51,8 @@ def search_leaderboard(df, query):
         return df[df['Method'].str.contains(query)]
 
 # Function to change the version of the leaderboard
-def change_version(model, version):
-    new_df = load_data(model, version)
+def change_version(model, version, metrics):
+    new_df = load_data(model, version, metrics)
     return new_df
 
 # Function to create plots
@@ -94,7 +94,7 @@ with demo:
                 )
 
             leaderboard_table = gr.components.Dataframe(
-                value=load_data("llama", "10%"),
+                value=load_data("llama", "10%", ["ROUGE", "Truth Ratio", "Prob."]),
                 interactive=True,
                 visible=True,
             )