Update app.py

app.py

@@ -9,12 +9,12 @@ def load_leaderboard():
 def display_data(data_type="Overall"):
     df = load_leaderboard()
     if data_type == "Overall":
-        new_df = df[["Model", "Overall Benign Score"]].copy()
+        new_df = df[["Model", "Overall Benign Score", "Overall Malicious Score", "Refusal Rate", "Normalized Safety Score"]].copy()
         sort_col = "Overall Benign Score"
     else:
-        new_df = df[["Model", data_type, "Overall Benign Score"]].copy()
-        new_df = new_df.rename(columns={data_type: "Category Score"})
-        sort_col = "Category Score"
+        category_col = f"{data_type} Score"
+        new_df = df[["Model", category_col, "Overall Benign Score"]].copy()
+        sort_col = category_col
     return new_df.sort_values(by=sort_col, ascending=False)
 
 with gr.Blocks() as demo:
@@ -34,4 +34,4 @@ with gr.Blocks() as demo:
         outputs=table
     )
     
-    demo.launch(share=True)
+    demo.launch(share=True)