devingulliver
commited on
Commit
•
2620c5c
1
Parent(s):
3a5e1ef
Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,6 @@ archlinks = {
|
|
21 |
def filter_table(cols, name, type, arch, size):
|
22 |
tmp = data
|
23 |
# filter
|
24 |
-
tmp = tmp[tmp["Name"] != "EleutherAI/pythia-1.4b"]
|
25 |
tmp = tmp[tmp["Name"].str.contains(name, case=False)]
|
26 |
tmp = tmp[tmp["Type"].isin(type)]
|
27 |
tmp = tmp[tmp["Architecture"].isin(arch)]
|
@@ -72,13 +71,13 @@ with gr.Blocks(css=".gradio-container{max-width:95%!important} .tab-buttons butt
|
|
72 |
typefilter = gr.CheckboxGroup(show_label=False, choices=list(data["Type"].unique()), value=[n for n in data["Type"].unique() if n not in ["⏳ Pending"]])
|
73 |
|
74 |
with gr.Column():
|
75 |
-
archfilter = gr.CheckboxGroup(label="Filter by model architecture", choices=list(data["Architecture"].unique()), value=list(data["Architecture"].unique()))
|
76 |
sizefilter = gr.CheckboxGroup(label="Filter by model size", choices=list(data["Model Size"].unique()), value=list(data["Model Size"].unique()))
|
77 |
|
78 |
with gr.Column():
|
79 |
colfilter = gr.CheckboxGroup(label="Hide columns", choices=list(data.columns)[2:], value=["Architecture","Model Size","Base Model"])
|
80 |
|
81 |
-
table = gr.Dataframe(filter_table(["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")
|
82 |
|
83 |
# actions
|
84 |
|
|
|
21 |
def filter_table(cols, name, type, arch, size):
|
22 |
tmp = data
|
23 |
# filter
|
|
|
24 |
tmp = tmp[tmp["Name"].str.contains(name, case=False)]
|
25 |
tmp = tmp[tmp["Type"].isin(type)]
|
26 |
tmp = tmp[tmp["Architecture"].isin(arch)]
|
|
|
71 |
typefilter = gr.CheckboxGroup(show_label=False, choices=list(data["Type"].unique()), value=[n for n in data["Type"].unique() if n not in ["⏳ Pending"]])
|
72 |
|
73 |
with gr.Column():
|
74 |
+
archfilter = gr.CheckboxGroup(label="Filter by model architecture", choices=[n for n in list(data["Architecture"].unique()) if n != "Transformer"], value=[n for n in list(data["Architecture"].unique()) if n != "Transformer"])
|
75 |
sizefilter = gr.CheckboxGroup(label="Filter by model size", choices=list(data["Model Size"].unique()), value=list(data["Model Size"].unique()))
|
76 |
|
77 |
with gr.Column():
|
78 |
colfilter = gr.CheckboxGroup(label="Hide columns", choices=list(data.columns)[2:], value=["Architecture","Model Size","Base Model"])
|
79 |
|
80 |
+
table = gr.Dataframe(filter_table(["Architecture","Model Size","Base Model"],"",[n for n in data["Type"].unique() if n not in ["⏳ Pending"]],[n for n in list(data["Architecture"].unique()) if n != "Transformer"],list(data["Model Size"].unique())), datatype="markdown")
|
81 |
|
82 |
# actions
|
83 |
|