Update app.py

app.py

@@ -182,6 +182,13 @@ def compare_models():
 
     results_df = pd.DataFrame(results)
     return results_df
+    
+# 1. prepare_data should come first
+def prepare_data(df, target_column="target"):
+    from sklearn.model_selection import train_test_split
+    X = df.drop(columns=[target_column])
+    y = df[target_column]
+    return train_test_split(X, y, test_size=0.2, random_state=42)
 
 def train_model(_):
     try: