fix

app.py

@@ -178,9 +178,12 @@ def display(x, y):
     for column in original_df.columns:
         if original_df[column].dtype == 'object':
             original_df[column] = original_df[column].apply(remove_invalid_unicode)
-    json_data = original_df.to_json(orient='records', columns=COLS)
+
+    subset_df = original_df[COLS]
+
+    json_data = subset_df.to_json(orient='records')
     print(json_data)  # Print JSON representation
-    return original_df
+    return subset_df
 
 INTRODUCTION_TEXT = """
 This is a copied space from Open Source LLM leaderboard. Instead of displaying