Update app.py

app.py

@@ -162,12 +162,14 @@ with gr.Blocks() as demo:
 
     def run(mode, n_points, noise, n_estimators, max_depth, show_trees, rotate):
         if mode == "2D Regression":
-            return (*rf_2d(n_points, noise, n_estimators, max_depth, show_trees))
+            fig1, fig2, text = rf_2d(n_points, noise, n_estimators, max_depth, show_trees)
+            return fig1, fig2, text
         elif mode == "3D Regression":
             frames, text = rf_3d(n_points, noise, n_estimators, max_depth, rotate)
             return frames[0], None, text
         else:
-            return (*rf_classification(n_points, noise, n_estimators, max_depth))
+            fig1, fig2, text = rf_classification(n_points, noise, n_estimators, max_depth)
+            return fig1, fig2, text
 
     inputs = [mode, n_points, noise, n_estimators, max_depth, show_trees, rotate]