Nathan Habib commited on
Commit
042ea78
1 Parent(s): 38b604c

checkbox for unkown model type

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -251,7 +251,7 @@ def filter_models(
251
  filtered_df = df[df[AutoEvalColumn.still_on_hub.name] == True]
252
 
253
  type_emoji = [t[0] for t in type_query]
254
- filtered_df = filtered_df[df[AutoEvalColumn.model_type_symbol.name].isin(type_emoji + ["?"])]
255
  filtered_df = filtered_df[df[AutoEvalColumn.precision.name].isin(precision_query + ["None"])]
256
 
257
  numeric_interval = pd.IntervalIndex(sorted([NUMERIC_INTERVALS[s] for s in size_query]))
 
251
  filtered_df = df[df[AutoEvalColumn.still_on_hub.name] == True]
252
 
253
  type_emoji = [t[0] for t in type_query]
254
+ filtered_df = filtered_df[df[AutoEvalColumn.model_type_symbol.name].isin(type_emoji)]
255
  filtered_df = filtered_df[df[AutoEvalColumn.precision.name].isin(precision_query + ["None"])]
256
 
257
  numeric_interval = pd.IntervalIndex(sorted([NUMERIC_INTERVALS[s] for s in size_query]))