Update model.py

src/models/model.py

@@ -75,3 +75,7 @@ class DataModule(Dataset):
             ] = -100
 
         return dict(
+            keywords=data_row["keywords"],
+            text=data_row["text"],
+            keywords_input_ids=input_encoding["input_ids"].flatten(),
+            keywords_attention_mask=input_encoding["attention_mask"].flatten(),