Update app.py

app.py

@@ -131,8 +131,10 @@ def update_table(df: pd.DataFrame, query: str, param_ranges: list, is_foundation
                 size_mask |= ((df['Total Params'] >= 28) & (df['Total Params'] < 40))
             elif param_range == '~50':
                 size_mask |= ((df['Total Params'] >= 40) & (df['Total Params'] < 65))
-            elif param_range == '~70+':
-                size_mask |= (df['Total Params'] >= 65)
+            elif param_range == '~70':
+                size_mask |= ((df['Total Params'] >= 65) & (df['Total Params'] < 90))
+            elif param_range == '~90+':
+                size_mask |= (df['Total Params'] >= 90)
             elif param_range == 'Closed':
                 size_mask |= df['Total Params'].isna()
         mask &= size_mask
@@ -188,7 +190,7 @@ with GraInter:
             with gr.Column(scale=7):
                 filter_columns_size = gr.CheckboxGroup(
                     label="Model sizes (in billions of parameters)",
-                    choices=['~2', '~4', '~8', '~13', '~20', '~34', '~50', '~70+', 'Closed'],
+                    choices=['~2', '~4', '~8', '~13', '~20', '~34', '~50', '~70', '~90+', 'Closed'],
                     value=[],
                     interactive=True,
                     elem_id="filter-columns-size",