djstrong commited on
Commit
8a3bdad
1 Parent(s): 45d02c6

fix search

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -116,7 +116,7 @@ def filter_queries(query: str, filtered_df: pd.DataFrame) -> pd.DataFrame:
116
  if len(final_df) > 0:
117
  filtered_df = pd.concat(final_df)
118
  filtered_df = filtered_df.drop_duplicates(
119
- subset=[AutoEvalColumn.model.name, AutoEvalColumn.precision.name, AutoEvalColumn.revision.name]
120
  )
121
 
122
  return filtered_df
 
116
  if len(final_df) > 0:
117
  filtered_df = pd.concat(final_df)
118
  filtered_df = filtered_df.drop_duplicates(
119
+ subset=[AutoEvalColumn.model.name, AutoEvalColumn.n_shot.name, AutoEvalColumn.precision.name, AutoEvalColumn.revision.name]
120
  )
121
 
122
  return filtered_df