fixes

src/models/model.py

@@ -76,8 +76,8 @@ class DataModule(Dataset):
             ] = -100
 
         return dict(
-            keywords=data_row["keywords"],
-            text=data_row["text"],
+            input_text=data_row["input_text"],
+            output_text=data_row["output_text"],
             keywords_input_ids=input_encoding["input_ids"].flatten(),
             keywords_attention_mask=input_encoding["attention_mask"].flatten(),
             labels=labels.flatten(),