Update app.py

app.py

@@ -67,9 +67,9 @@ def update_table(df: pd.DataFrame, query: str, param_ranges: list, columns: list
     if param_ranges:
         param_mask = pd.Series(False, index=filtered_df.index)
         for param_range in param_ranges:
-            if param_range == '~1.5':
+            if param_range == '~2':
                 param_mask |= (filtered_df['Params'] < 2.5)
-            elif param_range == '~3':
+            elif param_range == '~4':
                 param_mask |= ((filtered_df['Params'] >= 2.5) & (filtered_df['Params'] < 6))
             elif param_range == '~8':
                 param_mask |= ((filtered_df['Params'] >= 6) & (filtered_df['Params'] < 9.5))
@@ -115,7 +115,7 @@ with GraInter:
         with gr.Row():
             filter_columns_size = gr.CheckboxGroup(
                 label="Model sizes (in billions of parameters)",
-                choices=['~1.5', '~3', '~8', '~13', '~20', '~34', '~50', '~70+'],
+                choices=['~2', '~4', '~8', '~13', '~20', '~34', '~50', '~70+'],
                 value=[],
                 interactive=True,
                 elem_id="filter-columns-size",