Update model.py

src/models/model.py

@@ -15,6 +15,7 @@ from pytorch_lightning import Trainer
 from pytorch_lightning.callbacks.early_stopping import EarlyStopping
 from pytorch_lightning import LightningDataModule
 from pytorch_lightning import LightningModule
+from datasets import load_metric
 
 torch.cuda.empty_cache()
 pl.seed_everything(42)