Update model.py

src/models/model.py

@@ -2,6 +2,7 @@ import time
 import torch
 import numpy as np
 import pandas as pd
+from dagshub.pytorch_lightning import DAGsHubLogger
 from transformers import (
     AdamW,
     T5ForConditionalGeneration,