Fix seaborn color parameter

app.py

@@ -298,7 +298,7 @@ mT5 green and the other models black.
         y="en->nl Bleu",
         x="summ Rouge1",
         size="num params (M)",
-        color=colors,
+        hue=colors,
         linewidth=0.7,
     )
     for i, row in plot_df.iterrows():
@@ -311,6 +311,12 @@ mT5 green and the other models black.
             va="center",
             rotation=0,
         )
+    # Remove color legend
+    handles, labels = ax.get_legend_handles_labels()
+    size_legend_labels = ["num params (M)"] + labels[-4:]
+    size_legend_handles = handles[-5:]
+    ax.legend(handles=size_legend_handles, labels=size_legend_labels)
+
     plt.tight_layout()
     st.pyplot(fig)
     st.markdown(