Update app.py

app.py

@@ -68,7 +68,7 @@ def curve_fit(size, alpha, lam):
         )
         ax.text(0.05, -1.0, text, fontsize=12)
     
-    return axes[0], axes[1]
+    return fig[0], fig[1]
 
 
 with gr.Blocks(theme=theme) as demo: