DontPlanToEnd commited on
Commit
5dceefc
β€’
1 Parent(s): ac35267

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -15,7 +15,7 @@ with open(os.path.join(current_dir, "Images/kofi_button.png"), "rb") as image_fi
15
  # Create the HTML for the kofi button
16
  KOFI_BUTTON_HTML = f"""
17
  <a href="https://ko-fi.com/dontplantoend" target="_blank">
18
- <img src="data:image/png;base64,{kofi_button}" style="width:150px;display:block;margin-left:auto;margin-right:auto">
19
  </a>
20
  """
21
 
@@ -120,6 +120,8 @@ def update_table(df: pd.DataFrame, query: str, param_ranges: list, columns: list
120
  param_mask |= ((filtered_df['Total Params'] >= 40) & (filtered_df['Total Params'] < 65))
121
  elif param_range == '~70+':
122
  param_mask |= (filtered_df['Total Params'] >= 65)
 
 
123
  filtered_df = filtered_df[param_mask]
124
 
125
  if query:
@@ -165,10 +167,10 @@ with GraInter:
165
  with gr.Row():
166
  search_bar = gr.Textbox(placeholder=" πŸ” Search for a model...", show_label=False, elem_id="search-bar")
167
  with gr.Row():
168
- with gr.Column(scale=3):
169
  filter_columns_size = gr.CheckboxGroup(
170
  label="Model sizes (in billions of parameters)",
171
- choices=['~2', '~4', '~8', '~13', '~20', '~34', '~50', '~70+'],
172
  value=[],
173
  interactive=True,
174
  elem_id="filter-columns-size",
 
15
  # Create the HTML for the kofi button
16
  KOFI_BUTTON_HTML = f"""
17
  <a href="https://ko-fi.com/dontplantoend" target="_blank">
18
+ <img src="data:image/png;base64,{kofi_button}" style="width:165px;display:block;margin-left:auto;margin-right:auto">
19
  </a>
20
  """
21
 
 
120
  param_mask |= ((filtered_df['Total Params'] >= 40) & (filtered_df['Total Params'] < 65))
121
  elif param_range == '~70+':
122
  param_mask |= (filtered_df['Total Params'] >= 65)
123
+ elif param_range == 'Closed':
124
+ param_mask |= filtered_df['Total Params'].isna()
125
  filtered_df = filtered_df[param_mask]
126
 
127
  if query:
 
167
  with gr.Row():
168
  search_bar = gr.Textbox(placeholder=" πŸ” Search for a model...", show_label=False, elem_id="search-bar")
169
  with gr.Row():
170
+ with gr.Column(scale=4):
171
  filter_columns_size = gr.CheckboxGroup(
172
  label="Model sizes (in billions of parameters)",
173
+ choices=['~2', '~4', '~8', '~13', '~20', '~34', '~50', '~70+', 'Closed'],
174
  value=[],
175
  interactive=True,
176
  elem_id="filter-columns-size",