Update model.py

src/models/model.py

@@ -9,3 +9,4 @@ from transformers import (
     T5TokenizerFast as T5Tokenizer,
 )
 from torch.utils.data import Dataset, DataLoader
+import pytorch_lightning as pl