ubowang commited on
Commit
02d88d2
·
verified ·
1 Parent(s): 8475b2f

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -148,7 +148,7 @@ def search_and_filter_models(df, query, min_size, max_size):
148
  if query:
149
  df = df[df['Models'].str.contains(query, case=False, na=False)]
150
 
151
- df = df[(df['Model Size'] >= min_size) & (df['Model Size'] <= max_size)]
152
 
153
  return df
154
 
@@ -160,7 +160,7 @@ def search_models(df, query):
160
 
161
 
162
  def get_size_range(df):
163
- sizes = df['Model Size'].dropna()
164
  if len(sizes) > 0:
165
  return float(sizes.min()), float(sizes.max())
166
  return 0, 500
 
148
  if query:
149
  df = df[df['Models'].str.contains(query, case=False, na=False)]
150
 
151
+ df = df[(df['Model Size(B)'] >= min_size) & (df['Model Size(B)'] <= max_size)]
152
 
153
  return df
154
 
 
160
 
161
 
162
  def get_size_range(df):
163
+ sizes = df['Model Size(B)'].dropna()
164
  if len(sizes) > 0:
165
  return float(sizes.min()), float(sizes.max())
166
  return 0, 500