Tom Aarsen commited on
Commit
a812c3b
1 Parent(s): 8168003

Make search case insensitive

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -521,7 +521,7 @@ def filter_data(search_query, model_types, model_sizes, *full_dataframes):
521
  names = df["Model"].map(lambda x: re.match("<a .+?>(.+)</a>", x).group(1))
522
  masks = []
523
  for query in search_query.split(";"):
524
- masks.append(names.str.contains(query))
525
  df = df[reduce(lambda a, b: a | b, masks)]
526
 
527
  # Apply the model type filtering
 
521
  names = df["Model"].map(lambda x: re.match("<a .+?>(.+)</a>", x).group(1))
522
  masks = []
523
  for query in search_query.split(";"):
524
+ masks.append(names.str.lower().str.contains(query.lower()))
525
  df = df[reduce(lambda a, b: a | b, masks)]
526
 
527
  # Apply the model type filtering