Update app.py

app.py

@@ -68,9 +68,10 @@ def update_table(
     precision_query: str,
     size_query: list,
     show_deleted: bool,
+    italian_only: bool,
     query: str,
 ):
-    filtered_df = filter_models(hidden_df, type_query, size_query, precision_query, show_deleted)
+    filtered_df = filter_models(hidden_df, type_query, size_query, precision_query, show_deleted, italian_only)
     filtered_df = filter_queries(query, filtered_df)
     df = select_columns(filtered_df, columns)
     return df
@@ -112,7 +113,7 @@ def filter_queries(query: str, filtered_df: pd.DataFrame) -> pd.DataFrame:
 
 
 def filter_models(
-    df: pd.DataFrame, type_query: list, size_query: list, precision_query: list, show_deleted: bool
+    df: pd.DataFrame, type_query: list, size_query: list, precision_query: list, show_deleted: bool, italian_only: bool
 ) -> pd.DataFrame:
     # Show all models
     if show_deleted:
@@ -128,6 +129,8 @@ def filter_models(
     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]
+    if italian_only:
+        filtered_df = filtered_df[filtered_df[AutoEvalColumn.author.name] == "🇮🇹"]
 
     return filtered_df
 
@@ -174,7 +177,11 @@ with demo:
                             value=True, label="Show gated/private/deleted models", interactive=True
                         )
                 with gr.Column(min_width=320):
-                    #with gr.Box(elem_id="box-filter"):
+                    italian_only_checkbox = gr.Checkbox(
+                        label="Show only models made by Italians",
+                        value=False,
+                        interactive=True,
+                        )
                     filter_columns_type = gr.CheckboxGroup(
                         label="Model types",
                         choices=[t.to_str() for t in ModelType],
@@ -225,6 +232,7 @@ with demo:
                     filter_columns_precision,
                     filter_columns_size,
                     deleted_models_visibility,
+                    italian_only_checkbox,
                     search_bar,
                 ],
                 leaderboard_table,
@@ -239,6 +247,7 @@ with demo:
                         filter_columns_precision,
                         filter_columns_size,
                         deleted_models_visibility,
+                        italian_only_checkbox,
                         search_bar,
                     ],
                     leaderboard_table,