update figure dimensions

app.py

@@ -5,7 +5,7 @@ import seaborn as sns
 from matplotlib import pyplot as plt
 
 sns.set_style(style="darkgrid")
-sns.set_context(context="notebook", font_scale=1.2)
+sns.set_context(context="notebook", font_scale=0.6)
 
 MAX_NOISE = 20
 DEFAULT_NOISE = 6
@@ -33,7 +33,7 @@ def generate_equation(process_params):
     # Substitute parameter values into the expression
     polynomial_with_values = polynomial_expression.subs(parameters)
     latex_representation = sp.latex(polynomial_with_values)
-    return fr"$${latex_representation}$$"
+    return fr"Underlying process $${latex_representation}$$"
 
 
 def true_process(x, process_params):
@@ -72,33 +72,34 @@ def make_plot(
 
     x, x_dense, y, y_dense, y_noisy = generate_data(num_points, noise_level, process_params)
 
-    fig = plt.figure(figsize=(10, 10), dpi=300)
+    fig = plt.figure(dpi=300)
     if show_true_process:
         plt.plot(
             x_dense, y_dense, "-", color="#363A4F",
             label="True Process",
+            lw=1.5,
         )
     if show_added_noise:
         plt.vlines(
             x, y, y_noisy, color="#556D9A",
             linestyles="dashed",
             alpha=0.75,
-            lw=1.2,
+            lw=1,
             label="Added Noise",
         )
     if show_original_points:
         plt.plot(
             x, y, "-o", color="none",
-            ms=8,
+            ms=6,
             markerfacecolor="white",
             markeredgecolor="#556D9A",
-            markeredgewidth=1.5,
+            markeredgewidth=1.2,
             label="Original Points",
         )
     if show_noisy_points:
         plt.plot(
             x, y_noisy, "-o", color="none",
-            ms=8.5,
+            ms=6.5,
             markerfacecolor="#556D9A",
             markeredgecolor="none",
             markeredgewidth=1.5,
@@ -106,9 +107,9 @@ def make_plot(
             label="Noisy Points",
         )
 
-    plt.xlabel("\nX")
-    plt.ylabel("\nY") 
-    plt.legend(fontsize=11)
+    plt.xlabel("\nx")
+    plt.ylabel("\ny") 
+    plt.legend(fontsize=7.5)
     plt.tight_layout()
     plt.show()
     return fig
@@ -155,7 +156,7 @@ with gr.Blocks() as demo:
                     show_params.append(gr.Checkbox(label="Show Added Noise", value=True))
                     show_params.append(gr.Checkbox(label="Show Noisy Points", value=True))
 
-            scatter_plot = gr.Plot(scale=1)
+            scatter_plot = gr.Plot()
 
     num_points.change(fn=make_plot, inputs=[num_points, noise_level, process_params, *show_params], outputs=scatter_plot)
     noise_level.change(fn=make_plot, inputs=[num_points, noise_level, process_params, *show_params], outputs=scatter_plot)