Update model.py

src/models/model.py

@@ -33,3 +33,5 @@ class DataModule(Dataset):
             target_max_token_len: int = 512,
     ):
         """
+        :param data:
+        :param tokenizer: