Update src/streamlit_app.py

src/streamlit_app.py

@@ -163,8 +163,8 @@ def create_bar_chart(df, view_type):
     df_fig["model"] = df_fig["model"].map(model_renaming_map).fillna(df_fig["model"])
 
     # TODO Rescale the scores
-    good_score_threshold = df_fig[df_fig["model"] == "Human Expert"].mean()
-    bad_score_threshold = df_fig[df_fig["model"] == "Straight Forward"].mean()
+    good_score_threshold = df_fig[df_fig["model"] == "Human Expert"]["score"].mean()
+    bad_score_threshold = df_fig[df_fig["model"] == "Straight Forward"]["score"].mean()
     df_fig["score"] = (bad_score_threshold - df_fig["score"]) / (bad_score_threshold - good_score_threshold) * 100
     
     if view_type == "Total Score":