use query param only with demo.load

app.py

@@ -23,12 +23,16 @@ except Exception:
     restart_space()
 
 
-def plot_evals(model_id, plotly_mode, request: gr.Request):
-    df = df_cot_err.copy()
+def plot_evals_init(model_id, plotly_mode, request: gr.Request):
     if request and "model" in request.query_params:
         model_param = request.query_params["model"]
-        if model_param in df.model.to_list():
+        if model_param in df_cot_err.model.to_list():
             model_id = model_param
+    return plot_evals(model_id, plotly_mode)
+
+
+def plot_evals(model_id, plotly_mode):
+    df = df_cot_err.copy()
     df["selected"] = df_cot_err.model.apply(lambda x: "selected" if x==model_id else "-")
     #df.sort_values(["selected", "model"], inplace=True, ascending=True)  # has currently no effect with px.scatter
     template = "plotly_dark" if plotly_mode=="dark" else "plotly" 
@@ -45,7 +49,16 @@ def plot_evals(model_id, plotly_mode, request: gr.Request):
     )
     return fig, model_id
 
-def get_model_table(model_id):
+
+def styled_model_table_init(model_id, request: gr.Request):
+    if request and "model" in request.query_params:
+        model_param = request.query_params["model"]
+        if model_param in df_cot_regimes.model.to_list():
+            model_id = model_param
+    return styled_model_table(model_id)
+
+
+def styled_model_table(model_id):
 
     def make_pretty(styler):
         styler.hide(axis="index")
@@ -79,13 +92,6 @@ def get_model_table(model_id):
     
     return df_cot_model.style.pipe(make_pretty)
 
-def styled_model_table(model_id, request: gr.Request):
-    if request and "model" in request.query_params:
-        model_param = request.query_params["model"]
-        if model_param in df_cot_regimes.model.to_list():
-            model_id = model_param
-    return get_model_table(model_id)
-
 
 demo = gr.Blocks()
 
@@ -95,7 +101,7 @@ with demo:
     gr.Markdown(INTRODUCTION_TEXT)
     with gr.Row():
         model_list = gr.Dropdown(list(df_cot_err.model.unique()), value="allenai/tulu-2-70b", label="Model", scale=2)
-        plotly_mode = gr.Radio(["dark","light"], value="dark", label="Plot theme", scale=1)
+        plotly_mode = gr.Radio(["dark","light"], value="light", label="Plot theme", scale=1)
         submit = gr.Button("Update", scale=1)
     table = gr.DataFrame()
     plot = gr.Plot(label="evals")
@@ -103,7 +109,8 @@ with demo:
 
     submit.click(plot_evals, [model_list, plotly_mode], [plot, model_list])
     submit.click(styled_model_table, model_list, table)
-    demo.load(plot_evals, [model_list, plotly_mode], [plot, model_list])
-    demo.load(styled_model_table, model_list, table)
+
+    demo.load(plot_evals_init, [model_list, plotly_mode], [plot, model_list])
+    demo.load(styled_model_table_init, model_list, table)
 
 demo.launch()