Spaces:
Running
Running
Update components.py
Browse files- components.py +2 -2
components.py
CHANGED
@@ -73,7 +73,7 @@ def model_selector(models: list[str], data: pd.DataFrame) -> set[str]:
|
|
73 |
# Group models by their prefix
|
74 |
model_groups: dict[str, list[str]] = {}
|
75 |
for model in models:
|
76 |
-
group, model_name = model.split("
|
77 |
if group not in model_groups:
|
78 |
model_groups[group] = []
|
79 |
model_groups[group].append(model_name)
|
@@ -353,7 +353,7 @@ def models_view(data: pd.DataFrame):
|
|
353 |
with right:
|
354 |
st.markdown("#### Frameworks")
|
355 |
_df = models.copy()
|
356 |
-
_df["Framework"] = _df.index.str.split("
|
357 |
fig = px.pie(
|
358 |
_df.groupby("Framework").size().reset_index(),
|
359 |
values=0,
|
|
|
73 |
# Group models by their prefix
|
74 |
model_groups: dict[str, list[str]] = {}
|
75 |
for model in models:
|
76 |
+
group, model_name = model.split("-", maxsplit=1)
|
77 |
if group not in model_groups:
|
78 |
model_groups[group] = []
|
79 |
model_groups[group].append(model_name)
|
|
|
353 |
with right:
|
354 |
st.markdown("#### Frameworks")
|
355 |
_df = models.copy()
|
356 |
+
_df["Framework"] = _df.index.str.split("-").str[0]
|
357 |
fig = px.pie(
|
358 |
_df.groupby("Framework").size().reset_index(),
|
359 |
values=0,
|