Update app.py

app.py

@@ -132,6 +132,8 @@ def update_table(df: pd.DataFrame, query: str, param_ranges: list, columns: list
                 param_mask |= (filtered_df['Total Params'] >= 65)
             elif param_range == 'Closed':
                 param_mask |= filtered_df['Total Params'].isna()
+            elif param_range == 'Foundation':
+                param_mask |= (filtered_df['Foundation'] == 1)
         filtered_df = filtered_df[param_mask]
     
     if query:
@@ -177,10 +179,10 @@ with GraInter:
         with gr.Row():
             search_bar = gr.Textbox(placeholder=" 🔍 Search for a model...", show_label=False, elem_id="search-bar")
         with gr.Row():
-            with gr.Column(scale=4):
+            with gr.Column(scale=5):
                 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+', 'Closed', 'Foundation'],
                     value=[],
                     interactive=True,
                     elem_id="filter-columns-size",