Update model.py

src/models/model.py

@@ -255,6 +255,7 @@ class Summarization:
             model_name (str, optional): exact model architecture name, "t5-base" or "t5-large". Defaults to "t5-base".
             :param model_type:
         """
+        if model_type == "t5":
 
     def train(
             self,