Clémentine commited on
Commit
99b25b8
1 Parent(s): a0b557b

added selector for model type

Browse files
Files changed (2) hide show
  1. .gitignore +5 -1
  2. app.py +7 -1
.gitignore CHANGED
@@ -8,4 +8,8 @@ __pycache__/
8
 
9
  gpt_4_evals/
10
  human_evals/
11
- model_counts.html
 
 
 
 
 
8
 
9
  gpt_4_evals/
10
  human_evals/
11
+ eval-queue/
12
+ eval-results/
13
+ auto_evals/
14
+
15
+ src/assets/model_counts.html
app.py CHANGED
@@ -257,7 +257,13 @@ def refresh():
257
 
258
 
259
  def search_table(df, query):
260
- filtered_df = df[df[AutoEvalColumn.dummy.name].str.contains(query, case=False)]
 
 
 
 
 
 
261
  return filtered_df
262
 
263
 
 
257
 
258
 
259
  def search_table(df, query):
260
+ if AutoEvalColumn.model_type.name in df.columns:
261
+ filtered_df = df[
262
+ (df[AutoEvalColumn.dummy.name].str.contains(query, case=False))
263
+ | (df[AutoEvalColumn.model_type.name].str.contains(query, case=False))
264
+ ]
265
+ else:
266
+ filtered_df = df[(df[AutoEvalColumn.dummy.name].str.contains(query, case=False))]
267
  return filtered_df
268
 
269