Update model.py

src/models/model.py

@@ -361,7 +361,7 @@ class Summarization:
         trainer.fit(self.T5Model, self.data_module)
 
     def load_model(
-            self, model_type: str = 't5', model_dir: str = "../../models", use_gpu: bool = False
+            self, model_type: str = 't5', model_dir: str = "models", use_gpu: bool = False
     ):
         """
         loads a checkpoint for inferencing/prediction