Update app.py

app.py

@@ -41,6 +41,7 @@ def curve_fit(size, alpha, lam):
     reg = BayesianRidge(tol=1e-6, fit_intercept=False, compute_score=True)
 
     fig, axes = plt.subplots(1, 2, figsize=(8, 4))
+    results = []
     for i, ax in enumerate(axes):
     # Bayesian ridge regression with different initial value pairs
         if i == 0:
@@ -67,8 +68,9 @@ def curve_fit(size, alpha, lam):
             reg.alpha_, reg.lambda_, reg.scores_[-1]
         )
         ax.text(0.05, -1.0, text, fontsize=12)
+        results[i] = ax
     
-    return fig[0], fig[1]
+    return results
 
 
 with gr.Blocks(theme=theme) as demo: