Update app.py

app.py

@@ -7,12 +7,16 @@ import gradio as gr
 data = pd.read_csv("data.csv")
 webhook_url = os.environ.get("WEBHOOK_URL")
 
-def filter_table(name, type, arch, lcns):
+def filter_table(name, type, arch, license):
     tmp = data
-    tmp = tmp[tmp["Name"].str.contains(name)]
-    tmp = tmp[tmp["Type"].isin(type)]
-    tmp = tmp[tmp["Architecture"].isin(arch)]
-    tmp = tmp[tmp["License"].isin(lcns)]
+    if name:
+        tmp = tmp[tmp["Name"].str.contains(name)]
+    if type:
+        tmp = tmp[tmp["Type"].isin(type)]
+    if arch:
+        tmp = tmp[tmp["Architecture"].isin(arch)]
+    if license:
+        tmp = tmp[tmp["License"].isin(license)]
     return tmp
 
 def submit_model(name):
@@ -43,14 +47,15 @@ with gr.Blocks() as demo:
 
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
         with gr.Tab("🏅 LLM Benchmark"):
-            with gr.Row():
-                namefilter = model_name = gr.Textbox(max_lines=1, placeholder="Search by model name...", show_label=False)
-                typefilter = gr.Dropdown(label="Filter by model type", multiselect=True, choices=list(set(data["Type"])), value=list(set(data["Type"])))
-                archfilter = gr.Dropdown(label="Filter by model architecture", multiselect=True, choices=list(set(data["Architecture"])), value=list(set(data["Architecture"])))
-                lcnsfilter = gr.Dropdown(label="Filter by model license", multiselect=True, choices=list(set(data["License"])), value=list(set(data["License"])))
-                filter = gr.Button("Filter")
+            with gr.Group():
+                with gr.Row():
+                    namefilter = model_name = gr.Textbox(max_lines=1, placeholder="Search by model name...", show_label=False)
+                    typefilter = gr.Dropdown(label="Filter by model type", multiselect=True, choices=list(set(data["Type"])), value=["Base","Chat"])
+                    archfilter = gr.Dropdown(label="Filter by model architecture", multiselect=True, choices=list(set(data["Architecture"])))
+                    lcnsfilter = gr.Dropdown(label="Filter by model license", multiselect=True, choices=list(set(data["License"])))
+                    filter = gr.Button("Filter")
                 
-            table = gr.Dataframe(data)
+            table = gr.Dataframe(filter_table("",["Base","Chat"],[],[]))
 
             filter.click(fn=filter_table, inputs=[namefilter,typefilter,archfilter,lcnsfilter], outputs=table)