Update app.py

app.py

@@ -15,12 +15,12 @@ df.columns = df.columns.str.strip()  # Remove any whitespace from column names
 # Rename columns to match our expected format
 df = df.rename(columns={
     'model': 'Model Name',
-    'size': 'Size',
+    'size': 'Size (B)',
     "prompt": "Prompt"
 })
 
 # Create size display format
-df["Size_Display"] = df["Size"].apply(
+df["Size_Display"] = df["Size (B)"].apply(
     lambda x: f"{x}B" if x != "???" else f"???"
 )
 
@@ -47,7 +47,7 @@ def get_size_category(size):
         return ">80B"
 
 
-df["Size_Category"] = df["Size"].apply(get_size_category)
+df["Size_Category"] = df["Size (B)"].apply(get_size_category)
 
 
 def filter_and_search_models(