Update app.py

app.py

@@ -15,12 +15,13 @@ archlinks = {
     "StripedHyena": "https://www.together.ai/blog/stripedhyena-7b", # no paper?
 }
 
-def filter_table(cols, name, type, arch):
+def filter_table(cols, name, type, arch, size):
     tmp = data
     # filter
     tmp = tmp[tmp["Name"].str.contains(name)]
     tmp = tmp[tmp["Type"].isin(type)]
     tmp = tmp[tmp["Architecture"].isin(arch)]
+    tmp = tmp[tmp["Model Size"].isin(size)]
     # prettify
     tmp["Type"] = tmp["Type"].apply(lambda x: x[0])
     tmp = tmp.rename({"Type": "T"}, axis=1)
@@ -68,18 +69,19 @@ with gr.Blocks(css=".tab-buttons button{font-size:1.3em}") as demo:
                 
                 with gr.Column():
                     archfilter = gr.CheckboxGroup(label="Filter by model architecture", choices=list(data["Architecture"].unique()), value=list(data["Architecture"].unique()))
+                    sizefilter = gr.CheckboxGroup(label="Filter by model size", choices=list(data["Model Size"].unique()), value=list(data["Model Size"].unique()))
 
                 with gr.Column():
-                    colfilter = gr.CheckboxGroup(label="Hide columns", choices=list(data.columns)[2:], value=["MT-Bench (coming soon!)","Architecture","Base Model"])
+                    colfilter = gr.CheckboxGroup(label="Hide columns", choices=list(data.columns)[2:], value=["MT-Bench (coming soon!)","Architecture","Model Size","Base Model"])
                 
-            table = gr.Dataframe(filter_table(["MT-Bench (coming soon!)","Architecture","Base Model"],"",[n for n in data["Type"].unique() if n not in ["⌛ Pending"]],list(data["Architecture"].unique())), datatype="markdown")
+            table = gr.Dataframe(filter_table(["MT-Bench (coming soon!)","Architecture","Model Size","Base Model"],"",[n for n in data["Type"].unique() if n not in ["⌛ Pending"]],list(data["Architecture"].unique()),list(data["Model Size"].unique())), datatype="markdown")
 
             # actions
             
-            namefilter.submit(filter_table, [colfilter,namefilter,typefilter,archfilter], table)
+            namefilter.submit(filter_table, [colfilter,namefilter,typefilter,archfilter,sizefilter], table)
             
-            for filter in [colfilter,typefilter,archfilter]:
-                filter.input(filter_table, [colfilter,namefilter,typefilter,archfilter], table)
+            for filter in [colfilter,typefilter,archfilter,sizefilter]:
+                filter.input(filter_table, [colfilter,namefilter,typefilter,archfilter,sizefilter], table)
         
         with gr.Tab("📝 About"):
             gr.Markdown("""