Update app.py

app.py

@@ -70,7 +70,7 @@ def update_table(
     hidden_df: pd.DataFrame,
     columns: list,
     type_query: list,
-    precision_query: list,
+    precision_query: str,
     size_query: list,
     add_special_tokens_query: list,
     num_few_shots_query: list,
@@ -79,23 +79,12 @@ def update_table(
     show_flagged: bool,
     query: str,
 ):
-    print(f"Update table called with: type_query={type_query}, precision_query={precision_query}, size_query={size_query}")
-    print(f"hidden_df shape before filtering: {hidden_df.shape}")
-    
-    filtered_df = filter_models(hidden_df, type_query, size_query, precision_query,
-                                add_special_tokens_query, num_few_shots_query,
-                                show_deleted, show_merges, show_flagged)
-    print(f"filtered_df shape after filter_models: {filtered_df.shape}")
-    
-    if query:
-        filtered_df = filter_queries(query, filtered_df)
-        print(f"filtered_df shape after filter_queries: {filtered_df.shape}")
-    
+    filtered_df = filter_models(hidden_df, type_query, size_query, precision_query, add_special_tokens_query, num_few_shots_query, show_deleted, show_merges, show_flagged)
+    filtered_df = filter_queries(query, filtered_df)
     print(f"Filter applied: query={query}, columns={columns}, type_query={type_query}, precision_query={precision_query}")
-    print("Filtered dataframe head:")
-    print(filtered_df.head())
 
-    return select_columns(filtered_df, columns)
+    df = select_columns(filtered_df, columns)
+    return df
 
 
 def load_query(request: gr.Request):  # triggered only once at startup => read query parameter if it exists