fixes

src/models/model.py

@@ -335,9 +335,6 @@ class Summarization:
 
         # logger = DAGsHubLogger(metrics_path='reports/training_metrics.txt')
 
-        df = pd.read_json(r'wandb/latest-run/files/wandb-summary.json')
-        df.to_csv(r'reports/training_metrics.txt', index=False)
-
         early_stop_callback = (
             [
                 EarlyStopping(
@@ -364,6 +361,9 @@ class Summarization:
 
         trainer.fit(self.T5Model, self.data_module)
 
+        df = pd.read_json(r'wandb/latest-run/files/wandb-summary.json')
+        df.to_csv(r'reports/training_metrics.txt', index=False)
+
     def load_model(
             self, model_type: str = 't5', model_dir: str = "models", use_gpu: bool = False
     ):