Update preprocess.py

preprocess.py

@@ -42,8 +42,9 @@ def parse(csv_path):
     os.makedirs(output_folder, exist_ok=True)
     
     # Step 6: Save the train and test sets as CSV files
-    train_file_path = os.path.join(output_folder, "train.csv")
-    test_file_path = os.path.join(output_folder, "test.csv")
+    # train_file_path = os.path.join(output_folder, "train.csv")
+    train_file_path = "data/train.csv"
+    test_file_path = "data/test.csv"
     
     train_data.to_csv(train_file_path, index=False)
     test_data.to_csv(test_file_path, index=False)