Update app.py

app.py

@@ -170,6 +170,7 @@ def train_model(x, target_cols):
     performance_plot = cv2.imread("model_performance.png")
     metrics = pd.DataFrame([[0.05, 0.017]], columns=["RMSE", "MAPE"])
     next_df = x.sample(n=5, random_state=12)
+    next_df.drop(columns=target_cols, inplace=True)
     return "Model successfully adapted to your data!", performance_plot, metrics, next_df