Update app.py

app.py

@@ -9,7 +9,7 @@ from sklearn import linear_model
 from sklearn.svm import l1_min_c
 
 
-def train_it(solver, intersect_scaling, tol, max_iter):
+def train_it(solver, intersect_scaling, tol, max_iter,warm_start):
     iris = datasets.load_iris()
     X = iris.data
     y = iris.target
@@ -26,7 +26,7 @@ def train_it(solver, intersect_scaling, tol, max_iter):
         solver=solver,
         tol=tol,
         max_iter=int(max_iter),
-        warm_start=True,
+        warm_start=warm_start,
         intercept_scaling=intersect_scaling,
     )
 
@@ -86,7 +86,7 @@ with gr.Blocks() as demo:
     train_buttion = gr.Button(label="Train")
 
     train_buttion.click(
-        train_it, inputs=[solver, intersect_scaling, tol, max_iter], outputs=gr.Plot()
+        train_it, inputs=[solver, intersect_scaling, tol, max_iter,warm_start], outputs=gr.Plot()
     )