Fix the x-axis of the size vs. avg graph

app.py

@@ -257,6 +257,8 @@ INITIAL_COLUMNS = ["T"] + [
 ]
 leaderboard_df = select_columns(leaderboard_df, INITIAL_COLUMNS)
 
+MAX_MODEL_SIZE = original_df["#Params (B)"].max()
+
 
 # Leaderboard demo
 
@@ -318,7 +320,7 @@ def plot_size_vs_score(df_filtered: pd.DataFrame, df_original: pd.DataFrame) ->
         hovertemplate="<b>%{customdata[0]}</b><br>#Params: %{x:.2f}B<br>n-shot: %{customdata[1]}<br>AVG: %{y:.4f}<extra></extra>",
         textposition="top right",
     )
-    fig.update_layout(yaxis_range=[0, 1])
+    fig.update_layout(xaxis_range=[0, MAX_MODEL_SIZE * 1.2], yaxis_range=[0, 1])
     return fig