Update model.py

src/models/model.py

@@ -366,6 +366,7 @@ class Summarization:
             )
         elif model_type == "byt5":
             self.tokenizer = ByT5Tokenizer.from_pretrained(f"{model_dir}")
+            self.model = T5ForConditionalGeneration.from_pretrained(
 
         if use_gpu:
             if torch.cuda.is_available():