Update model.py

src/models/model.py

@@ -10,3 +10,4 @@ from transformers import (
 )
 from torch.utils.data import Dataset, DataLoader
 import pytorch_lightning as pl
+from pytorch_lightning.loggers import MLFlowLogger