Update model.py

src/models/model.py

@@ -85,3 +85,11 @@ class DataModule(Dataset):
 
 
 class PLDataModule(LightningDataModule):
+    def __init__(
+            self,
+            train_df: pd.DataFrame,
+            test_df: pd.DataFrame,
+            tokenizer: T5Tokenizer,
+            source_max_token_len: int = 512,
+            target_max_token_len: int = 512,
+            batch_size: int = 4,