devingulliver commited on
Commit
2620c5c
1 Parent(s): 3a5e1ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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