Tom Aarsen commited on
Commit
2db25dc
1 Parent(s): fefaea6

Base "Open" models on the proprietary model list

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2025,7 +2025,7 @@ def filter_data(search_query, model_types, model_sizes, *full_dataframes):
2025
  masks = []
2026
  for model_type in model_types:
2027
  if model_type == "Open":
2028
- masks.append(df["Model Size (Million Parameters)"] != "")
2029
  elif model_type == "Proprietary":
2030
  masks.append(df["Model"].isin(PROPRIETARY_MODELS))
2031
  elif model_type == "Sentence Transformers":
 
2025
  masks = []
2026
  for model_type in model_types:
2027
  if model_type == "Open":
2028
+ masks.append(~df["Model"].isin(PROPRIETARY_MODELS))
2029
  elif model_type == "Proprietary":
2030
  masks.append(df["Model"].isin(PROPRIETARY_MODELS))
2031
  elif model_type == "Sentence Transformers":