Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
4c60028
1
Parent(s):
cdce7a5
Update train_model.py
Browse files
src/models/train_model.py
CHANGED
@@ -12,8 +12,8 @@ def train_model():
|
|
12 |
params = yaml.safe_load(f)
|
13 |
|
14 |
# Load the data
|
15 |
-
train_df = pd.read_csv('
|
16 |
-
eval_df = pd.read_csv('
|
17 |
|
18 |
model = Summarization()
|
19 |
model.from_pretrained('t5','t5-base')
|
|
|
12 |
params = yaml.safe_load(f)
|
13 |
|
14 |
# Load the data
|
15 |
+
train_df = pd.read_csv('data/processed/train.csv')
|
16 |
+
eval_df = pd.read_csv('data/processed/validation.csv')
|
17 |
|
18 |
model = Summarization()
|
19 |
model.from_pretrained('t5','t5-base')
|