Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Tom Aarsen
commited on
Commit
•
485f27b
1
Parent(s):
5613607
Fix proprietary models disappearing after model size toggling
Browse files
app.py
CHANGED
@@ -2021,7 +2021,7 @@ def filter_data(search_query, model_types, model_sizes, *full_dataframes):
|
|
2021 |
df = df[reduce(lambda a, b: a | b, masks)]
|
2022 |
|
2023 |
# Apply the model type filtering
|
2024 |
-
if model_types != MODEL_TYPES:
|
2025 |
masks = []
|
2026 |
for model_type in model_types:
|
2027 |
if model_type == "Open":
|
@@ -2036,7 +2036,7 @@ def filter_data(search_query, model_types, model_sizes, *full_dataframes):
|
|
2036 |
df = pd.DataFrame(columns=df.columns)
|
2037 |
|
2038 |
# Apply the model size filtering
|
2039 |
-
if model_sizes !=
|
2040 |
numeric_interval = pd.IntervalIndex(sorted([NUMERIC_INTERVALS[model_size] for model_size in model_sizes]))
|
2041 |
sizes = df["Model Size (Million Parameters)"].replace('', 0)
|
2042 |
mask = sizes.apply(lambda size: any(numeric_interval.contains(size)))
|
|
|
2021 |
df = df[reduce(lambda a, b: a | b, masks)]
|
2022 |
|
2023 |
# Apply the model type filtering
|
2024 |
+
if set(model_types) != set(MODEL_TYPES):
|
2025 |
masks = []
|
2026 |
for model_type in model_types:
|
2027 |
if model_type == "Open":
|
|
|
2036 |
df = pd.DataFrame(columns=df.columns)
|
2037 |
|
2038 |
# Apply the model size filtering
|
2039 |
+
if set(model_sizes) != set(NUMERIC_INTERVALS.keys()):
|
2040 |
numeric_interval = pd.IntervalIndex(sorted([NUMERIC_INTERVALS[model_size] for model_size in model_sizes]))
|
2041 |
sizes = df["Model Size (Million Parameters)"].replace('', 0)
|
2042 |
mask = sizes.apply(lambda size: any(numeric_interval.contains(size)))
|