Update utils.py

utils.py

@@ -148,10 +148,16 @@ def refresh_data():
 def search_and_filter_models(df, query, min_size, max_size):
     if query:
         df = df[df['Models'].str.contains(query, case=False, na=False)]
+
+    numeric_mask = df['Model Size(B)'].apply(lambda x: isinstance(x, (int, float)))
+    size_filtered = df[numeric_mask & 
+                         (df['Model Size(B)'] >= min_size) & 
+                         (df['Model Size(B)'] <= max_size)]
+    unknown_entries = df[df['Model Size(B)'] == 'unknown']
+    return pd.concat([size_filtered, unknown_entries])[COLUMN_NAMES]
+    # df = df[(df['Model Size(B)'] >= min_size) & (df['Model Size(B)'] <= max_size)]
     
-    df = df[(df['Model Size(B)'] >= min_size) & (df['Model Size(B)'] <= max_size)]
-    
-    return df[COLUMN_NAMES]
+    # return df[COLUMN_NAMES]
 
 
 def search_models(df, query):
@@ -169,9 +175,9 @@ def get_size_range(df):
 
 def process_model_size(size):
     if size == 'unk':
-        return 1000.0
+        return 'unknown'
     try:
         return float(size)
     except ValueError:
-        return 1000.0
+        return 'unknown'