fix bug

Signed-off-by: n1ck-guo <[email protected]>

app.py

@@ -386,22 +386,15 @@ def filter_models(
 
     filtered_df = filtered_df.loc[df[AutoEvalColumn.group_size.name].isin(group_dtype)]
 
-    print(filtered_df['model_name_for_query'].to_list())
-
     numeric_interval = pd.IntervalIndex(sorted([NUMERIC_INTERVALS[s] for s in size_query]))
     params_column = pd.to_numeric(df[AutoEvalColumn.params.name], errors="coerce")
     mask = params_column.apply(lambda x: any(numeric_interval.contains(x)))
     filtered_df = filtered_df.loc[mask]
-    
-    print('step 2')
-    print(filtered_df['model_name_for_query'].to_list())
 
     numeric_interval_params = pd.IntervalIndex(sorted([NUMERIC_MODELSIZE[s] for s in params_query]))
     params_column_params = pd.to_numeric(df[AutoEvalColumn.model_size.name], errors="coerce")
     mask_params = params_column_params.apply(lambda x: any(numeric_interval_params.contains(x)))
     filtered_df = filtered_df.loc[mask_params]
-    print('step 3')
-    print(filtered_df['model_name_for_query'])
 
     return filtered_df
 

src/display/utils.py

@@ -382,5 +382,5 @@ NUMERIC_MODELSIZE = {
     "~36": pd.Interval(16, 36, closed="right"),
     "~48": pd.Interval(36, 48, closed="right"),
     "~64": pd.Interval(48, 64, closed="right"),
-    "~72": pd.Interval(64, 80, closed="right"),
+    ">72": pd.Interval(64, 200, closed="right"),
 }