TimWindecker commited on
Commit
5338988
·
verified ·
1 Parent(s): 4c7e655

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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()