zhuohan-7 commited on
Commit
75ecf86
1 Parent(s): d8d6447

Upload folder using huggingface_hub

Browse files
app/__pycache__/draw_diagram.cpython-310.pyc CHANGED
Binary files a/app/__pycache__/draw_diagram.cpython-310.pyc and b/app/__pycache__/draw_diagram.cpython-310.pyc differ
 
app/__pycache__/pages.cpython-310.pyc CHANGED
Binary files a/app/__pycache__/pages.cpython-310.pyc and b/app/__pycache__/pages.cpython-310.pyc differ
 
app/draw_diagram.py CHANGED
@@ -53,11 +53,13 @@ def draw(folder_name, category_one, category_two, sort, num_sort, model_size_ran
53
  # How to work on the model size range, filter the ones that are not in the range
54
  if model_size_range != 'All':
55
  if model_size_range == '<10B':
56
- chart_data = chart_data[chart_data['model_size'].astype(int) < 10]
57
  elif model_size_range == '10B-30B':
58
- chart_data = chart_data[(chart_data['model_size'].astype(int) >= 10) & (chart_data['model_size'].astype(int) < 30)]
59
  elif model_size_range == '>30B':
60
- chart_data = chart_data[chart_data['model_size'].astype(int) >= 30]
 
 
61
 
62
  models = st.multiselect("Please choose the model",
63
  sorted(chart_data['model_show'].tolist()),
 
53
  # How to work on the model size range, filter the ones that are not in the range
54
  if model_size_range != 'All':
55
  if model_size_range == '<10B':
56
+ chart_data = chart_data[chart_data['model_size'].astype(float) < 10]
57
  elif model_size_range == '10B-30B':
58
+ chart_data = chart_data[(chart_data['model_size'].astype(float) >= 10) & (chart_data['model_size'].astype(float) < 30)]
59
  elif model_size_range == '>30B':
60
+ chart_data = chart_data[chart_data['model_size'].astype(float) >= 30]
61
+
62
+ chart_data.drop(columns=['model_size'], inplace=True)
63
 
64
  models = st.multiselect("Please choose the model",
65
  sorted(chart_data['model_show'].tolist()),