Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Tom Aarsen
commited on
Commit
•
2eb890d
1
Parent(s):
d81785e
Introduce new intervals: 100M & 250M and 250M & 500M
Browse files
app.py
CHANGED
@@ -1863,7 +1863,8 @@ def update_url_language(event: gr.SelectData, current_task_language: dict, langu
|
|
1863 |
|
1864 |
NUMERIC_INTERVALS = {
|
1865 |
"<100M": pd.Interval(0, 100, closed="right"),
|
1866 |
-
">100M, <
|
|
|
1867 |
">500M, <1B": pd.Interval(500, 1000, closed="right"),
|
1868 |
">1B": pd.Interval(1000, 1_000_000, closed="right"),
|
1869 |
}
|
@@ -1897,17 +1898,11 @@ def filter_data(search_query, model_types, model_sizes, *full_dataframes):
|
|
1897 |
df = df[reduce(lambda a, b: a | b, masks)]
|
1898 |
|
1899 |
# Apply the model size filtering
|
1900 |
-
if model_sizes !=
|
1901 |
-
masks = []
|
1902 |
-
# Handle the ? only
|
1903 |
-
if "?" in model_sizes:
|
1904 |
-
masks.append(df["Model Size (Million Parameters)"] == "")
|
1905 |
-
model_sizes.remove("?")
|
1906 |
-
# Handle the numeric intervals only
|
1907 |
numeric_interval = pd.IntervalIndex(sorted([NUMERIC_INTERVALS[model_size] for model_size in model_sizes]))
|
1908 |
sizes = df["Model Size (Million Parameters)"].replace('', 0)
|
1909 |
-
|
1910 |
-
df = df[
|
1911 |
|
1912 |
output_dataframes.append(df)
|
1913 |
return output_dataframes
|
@@ -1937,8 +1932,8 @@ with gr.Blocks(css=css) as block:
|
|
1937 |
)
|
1938 |
filter_model_sizes = gr.CheckboxGroup(
|
1939 |
label="Model sizes (in number of parameters)",
|
1940 |
-
choices=
|
1941 |
-
value=
|
1942 |
interactive=True,
|
1943 |
elem_classes=["filter-checkbox-group"]
|
1944 |
)
|
|
|
1863 |
|
1864 |
NUMERIC_INTERVALS = {
|
1865 |
"<100M": pd.Interval(0, 100, closed="right"),
|
1866 |
+
">100M, <250M": pd.Interval(100, 250, closed="right"),
|
1867 |
+
">250M, <500M": pd.Interval(250, 500, closed="right"),
|
1868 |
">500M, <1B": pd.Interval(500, 1000, closed="right"),
|
1869 |
">1B": pd.Interval(1000, 1_000_000, closed="right"),
|
1870 |
}
|
|
|
1898 |
df = df[reduce(lambda a, b: a | b, masks)]
|
1899 |
|
1900 |
# Apply the model size filtering
|
1901 |
+
if model_sizes != list(NUMERIC_INTERVALS.keys()):
|
|
|
|
|
|
|
|
|
|
|
|
|
1902 |
numeric_interval = pd.IntervalIndex(sorted([NUMERIC_INTERVALS[model_size] for model_size in model_sizes]))
|
1903 |
sizes = df["Model Size (Million Parameters)"].replace('', 0)
|
1904 |
+
mask = sizes.apply(lambda size: any(numeric_interval.contains(size)))
|
1905 |
+
df = df[mask]
|
1906 |
|
1907 |
output_dataframes.append(df)
|
1908 |
return output_dataframes
|
|
|
1932 |
)
|
1933 |
filter_model_sizes = gr.CheckboxGroup(
|
1934 |
label="Model sizes (in number of parameters)",
|
1935 |
+
choices=list(NUMERIC_INTERVALS.keys()),
|
1936 |
+
value=list(NUMERIC_INTERVALS.keys()),
|
1937 |
interactive=True,
|
1938 |
elem_classes=["filter-checkbox-group"]
|
1939 |
)
|