Update model.py

src/models/model.py

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