Update app.py

app.py

@@ -168,7 +168,8 @@ def train_model(x, target_cols):
     time.sleep(6)
     performance_plot = cv2.imread("model_performance.png")
     metrics = pd.DataFrame([[0.05, 0.017]], columns=["RMSE", "MAPE"])
-    return "Model successfully adapted to your data!", performance_plot, metrics
+    next_df = x.sample(n=5, random_state=12)
+    return "Model successfully adapted to your data!", performance_plot, metrics, next_df
 
 
 
@@ -228,6 +229,7 @@ with gr.Blocks(css=css_styling, title=page_title, theme=osium_theme) as demo:
                 output_text = gr.Textbox(label="Training results")
                 output_plot = gr.Image(label="Training performance", elem_classes="output-image")
                 output_performance = gr.DataFrame(label="Model performance")
+                output_next_experiments = gr.DataFrame(label="Suggested experiments to improve performance")
 
     with gr.Tab(label="Run your model"):
         with gr.Row():
@@ -391,7 +393,7 @@ with gr.Blocks(css=css_styling, title=page_title, theme=osium_theme) as demo:
     train_button.click(
         fn=train_model,
         inputs=[input_csv, target_columns],
-        outputs=[output_text, output_plot, output_performance],
+        outputs=[output_text, output_plot, output_performance, output_next_experiments],
         show_progress=True,
     )