Update model.py

src/models/model.py

@@ -296,20 +296,20 @@ class Summarization:
             )
 
     def train(
-            self,
-            train_df: pd.DataFrame,
-            eval_df: pd.DataFrame,
-            source_max_token_len: int = 512,
-            target_max_token_len: int = 512,
-            batch_size: int = 8,
-            max_epochs: int = 5,
-            use_gpu: bool = True,
-            outputdir: str = "models",
-            early_stopping_patience_epochs: int = 0,  # 0 to disable early stopping feature
-            learning_rate: float = 0.0001,
-            adam_epsilon: float = 0.01,
-            num_workers: int = 2,
-            weight_decay: float = 0.0001
+        self,
+        train_df: pd.DataFrame,
+        eval_df: pd.DataFrame,
+        source_max_token_len: int = 512,
+        target_max_token_len: int = 512,
+        batch_size: int = 8,
+        max_epochs: int = 5,
+        use_gpu: bool = True,
+        outputdir: str = "models",
+        early_stopping_patience_epochs: int = 0,  # 0 to disable early stopping feature
+        learning_rate: float = 0.0001,
+        adam_epsilon: float = 0.01,
+        num_workers: int = 2,
+        weight_decay: float = 0.0001,
     ):
         """
         trains T5/MT5 model on custom dataset