Corey Morris commited on
Commit
7b77065
1 Parent(s): a450af5

Added a default model to compare

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -278,7 +278,11 @@ st.write("""
278
  examining the closest models reveals variations in performance on individual tasks.
279
  Such an analysis can uncover specific strengths and weaknesses and guide further exploration and improvement.
280
  """)
281
- selected_model_name = st.selectbox("Select a Model:", filtered_data.index.tolist())
 
 
 
 
282
 
283
  # Get the closest 5 models with unique indices
284
  closest_models_diffs = filtered_data['MMLU_average'].sub(filtered_data.loc[selected_model_name, 'MMLU_average']).abs()
 
278
  examining the closest models reveals variations in performance on individual tasks.
279
  Such an analysis can uncover specific strengths and weaknesses and guide further exploration and improvement.
280
  """)
281
+
282
+ default_model_name = "GPT-JT-6B-v0"
283
+
284
+ default_model_index = filtered_data.index.tolist().index(default_model_name) if default_model_name in filtered_data.index else 0
285
+ selected_model_name = st.selectbox("Select a Model:", filtered_data.index.tolist(), index=default_model_index)
286
 
287
  # Get the closest 5 models with unique indices
288
  closest_models_diffs = filtered_data['MMLU_average'].sub(filtered_data.loc[selected_model_name, 'MMLU_average']).abs()