gagan3012 commited on
Commit
84814bd
·
1 Parent(s): 9ac4dd2
Files changed (1) hide show
  1. src/models/model.py +2 -2
src/models/model.py CHANGED
@@ -76,8 +76,8 @@ class DataModule(Dataset):
76
  ] = -100
77
 
78
  return dict(
79
- input_text=data_row["input_text"],
80
- output_text=data_row["output_text"],
81
  keywords_input_ids=input_encoding["input_ids"].flatten(),
82
  keywords_attention_mask=input_encoding["attention_mask"].flatten(),
83
  labels=labels.flatten(),
 
76
  ] = -100
77
 
78
  return dict(
79
+ keywords=data_row["input_text"],
80
+ text=data_row["output_text"],
81
  keywords_input_ids=input_encoding["input_ids"].flatten(),
82
  keywords_attention_mask=input_encoding["attention_mask"].flatten(),
83
  labels=labels.flatten(),