Update app.py

app.py

@@ -81,8 +81,8 @@ def get_model_info(df):
 
     return df
 
-
 def calculate_highest_combined_score(data, column):
+    score_columns = ['Average', 'AGIEval', 'GPT4All', 'TruthfulQA', 'Bigbench']
     # Ensure the column exists and has numeric data
     if column not in data.columns or not pd.api.types.is_numeric_dtype(data[column]):
         return column, {}
@@ -100,7 +100,7 @@ def calculate_highest_combined_score(data, column):
 # Function to display the results of the highest combined scores
 def display_highest_combined_scores(data):
     with st.spinner('Calculating highest combined scores...'):
-        results = [calculate_highest_combined_score(data, col) for col in columns]
+        results = [calculate_highest_combined_score(data, col) for col in score_columns]
         for column, top_combinations in results:
             st.subheader(f"Top Combinations for {column}")
             for r, combinations in top_combinations.items():