Update src/streamlit_app.py
Browse files- src/streamlit_app.py +10 -12
src/streamlit_app.py
CHANGED
|
@@ -151,12 +151,18 @@ def validate_tsv_format(uploaded_file):
|
|
| 151 |
|
| 152 |
def create_bar_chart(df, view_type):
|
| 153 |
"""Create interactive bar chart based on view type"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 154 |
|
| 155 |
if view_type == "Total Score":
|
| 156 |
-
|
| 157 |
-
# Copy df
|
| 158 |
-
df_fig = df.copy()
|
| 159 |
-
df_fig = df_fig[df_fig["score"] != np.inf]
|
| 160 |
|
| 161 |
# Calculate mean score per model
|
| 162 |
df_fig = df_fig.groupby("model")[["score"]].mean().reset_index()
|
|
@@ -211,10 +217,6 @@ def create_bar_chart(df, view_type):
|
|
| 211 |
|
| 212 |
elif view_type == "Per Embodiment":
|
| 213 |
|
| 214 |
-
# Copy df
|
| 215 |
-
df_fig = df.copy()
|
| 216 |
-
df_fig = df_fig[df_fig["score"] != np.inf]
|
| 217 |
-
|
| 218 |
# Calculate the model order
|
| 219 |
df_model_order = df_fig.groupby("model")[["score"]].mean().reset_index()
|
| 220 |
model_order = df_model_order.sort_values(by="score", ascending=True)["model"].tolist()
|
|
@@ -292,10 +294,6 @@ def create_bar_chart(df, view_type):
|
|
| 292 |
)
|
| 293 |
|
| 294 |
else: # Per Category
|
| 295 |
-
|
| 296 |
-
# Copy df
|
| 297 |
-
df_fig = df.copy()
|
| 298 |
-
df_fig = df_fig[df_fig["score"] != np.inf]
|
| 299 |
|
| 300 |
# Calculate the model order
|
| 301 |
df_model_order = df_fig.groupby("model")[["score"]].mean().reset_index()
|
|
|
|
| 151 |
|
| 152 |
def create_bar_chart(df, view_type):
|
| 153 |
"""Create interactive bar chart based on view type"""
|
| 154 |
+
|
| 155 |
+
# Copy df
|
| 156 |
+
df_fig = df.copy()
|
| 157 |
+
df_fig = df_fig[df_fig["score"] != np.inf]
|
| 158 |
+
|
| 159 |
+
# Split too long names
|
| 160 |
+
model_renaming_map = {
|
| 161 |
+
"Qwen 3 VL 235b Thinking": "Qwen 3 VL<br>235b Thinking",
|
| 162 |
+
}
|
| 163 |
+
df_fig["model"] = df_fig["model"].map(model_renaming_map).fillna(df_fig["model"])
|
| 164 |
|
| 165 |
if view_type == "Total Score":
|
|
|
|
|
|
|
|
|
|
|
|
|
| 166 |
|
| 167 |
# Calculate mean score per model
|
| 168 |
df_fig = df_fig.groupby("model")[["score"]].mean().reset_index()
|
|
|
|
| 217 |
|
| 218 |
elif view_type == "Per Embodiment":
|
| 219 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 220 |
# Calculate the model order
|
| 221 |
df_model_order = df_fig.groupby("model")[["score"]].mean().reset_index()
|
| 222 |
model_order = df_model_order.sort_values(by="score", ascending=True)["model"].tolist()
|
|
|
|
| 294 |
)
|
| 295 |
|
| 296 |
else: # Per Category
|
|
|
|
|
|
|
|
|
|
|
|
|
| 297 |
|
| 298 |
# Calculate the model order
|
| 299 |
df_model_order = df_fig.groupby("model")[["score"]].mean().reset_index()
|