Update train_model.py

src/models/train_model.py

@@ -14,6 +14,7 @@ def train_model():
         params = yaml.safe_load(f)
 
     # Load the data
+    train_df = pd.read_csv("data/processed/train.csv")
 
     train_df = train_df.sample(frac=params['split'], replace=True, random_state=1)
     eval_df = eval_df.sample(frac=params['split'], replace=True, random_state=1)