Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 25, 2021

Commit

4c60028

•

1 Parent(s): cdce7a5

Update train_model.py

Files changed (1) hide show

src/models/train_model.py CHANGED Viewed

@@ -12,8 +12,8 @@ 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')
     model = Summarization()
     model.from_pretrained('t5','t5-base')

         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')
     model = Summarization()
     model.from_pretrained('t5','t5-base')