Update app.py

app.py

@@ -107,11 +107,26 @@ def show_results_tab(df):
     
 
     with gr.Row():
-        model_name = gr.Textbox(
-            value='Input the Model Name', 
-            label='Search Model Name',
-            interactive=True
-        )
+        with gr.Column():
+            model_name = gr.Textbox(
+                value='Input the Model Name', 
+                label='Search Model Name',
+                interactive=True
+            )
+        with gr.Column():
+            size_filter = gr.CheckboxGroup(
+                choices=MODEL_SIZE,
+                value=MODEL_SIZE,
+                label='Model Size',
+                interactive=True,
+            )
+        with gr.Column():
+            type_filter = gr.CheckboxGroup(
+                choices=MODEL_TYPE,
+                value=MODEL_TYPE,
+                label='Model Type',
+                interactive=True,
+            )
 
     with gr.Column():
         table = gr.DataFrame(
@@ -121,11 +136,23 @@ def show_results_tab(df):
                 column_widths=calculate_column_widths(df),
         )
 
+    
     model_name.submit(
         fn=filter_df,
         inputs=model_name,
         outputs=table
     )
+    size_filter.change(
+        fn=update_table,
+        inputs=[size_filter, type_filter],
+        outputs=table,
+    )
+    type_filter.change(
+        fn=update_table,
+        inputs=[size_filter, type_filter],
+        outputs=table,
+    )
+