Update process_data.py

src/data/process_data.py

@@ -8,7 +8,6 @@ def process_data(split="train"):
     with open("params.yml") as f:
         params = yaml.safe_load(f)
 
-    df.columns = ['Unnamed: 0', 'input_text', 'output_text']
     df = df.sample(frac=params['split'], replace=True, random_state=1)
     if os.path.exists("data/raw/{}.csv".format(split)):
         os.remove("data/raw/{}.csv".format(split))