Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 31, 2021

Commit

737db13

1 Parent(s): 56499fc

Update train_model.py

Files changed (1) hide show

src/models/train_model.py CHANGED Viewed

@@ -14,7 +14,6 @@ def train_model():
         params = yaml.safe_load(f)
     # Load the data
-    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)

         params = yaml.safe_load(f)
     # Load the data
     eval_df = pd.read_csv('data/processed/validation.csv')
     train_df = train_df.sample(frac=params['split'], replace=True, random_state=1)