Update app.py

app.py

@@ -3,9 +3,9 @@ from utils import *
 global data_component
 
 
-def update_table(query):
+def update_table(query, min_size, max_size):
     df = get_df()
-    filtered_df = search_models(df, query)
+    filtered_df = search_and_filter_models(df, query, min_size, max_size)
     return filtered_df
 
 
@@ -28,11 +28,23 @@ with gr.Blocks() as block:
                 TABLE_INTRODUCTION
             )
 
-            search_bar = gr.Textbox(
-                placeholder="Search models...",
-                show_label=False,
-                elem_id="search-bar"
-            )
+            with gr.Row():
+                search_bar = gr.Textbox(
+                    placeholder="Search models...",
+                    show_label=False,
+                    elem_id="search-bar"
+                )
+            df = get_df()
+            min_size, max_size = get_size_range(df)
+
+            with gr.Row():
+                size_slider = gr.RangeSlider(
+                    minimum=min_size,
+                    maximum=max_size,
+                    value=[min_size, max_size],
+                    step=0.1,
+                    label="Select the number of parameters (B)",
+                )
 
             data_component = gr.components.Dataframe(
                 value=get_df(),
@@ -44,8 +56,11 @@ with gr.Blocks() as block:
                 height=1000
             )
             refresh_button = gr.Button("Refresh")
-            search_bar.change(fn=update_table, inputs=[search_bar], outputs=data_component)
+            search_bar.change(fn=update_table, inputs=[search_bar, size_slider], outputs=data_component)
+            size_slider.change(fn=update_table, inputs=[search_bar, size_slider], outputs=data_component)
             refresh_button.click(fn=refresh_data, outputs=data_component)
+            # search_bar.change(fn=update_table, inputs=[search_bar], outputs=data_component)
+            # refresh_button.click(fn=refresh_data, outputs=data_component)
 
         # table 2
         with gr.TabItem("📝 About", elem_id="qa-tab-table2", id=2):