Update train_model.py

src/models/train_model.py

@@ -17,6 +17,7 @@ def train_model():
     train_df = pd.read_csv("data/processed/train.csv")
     eval_df = pd.read_csv("data/processed/validation.csv")
 
+    train_df = train_df.sample(frac=params["split"], replace=True, random_state=1)
 
     model = Summarization()
     model.from_pretrained(model_type=params['model_type'], model_name=params['model_name'])