fix

src/Surveyor.py

@@ -139,14 +139,7 @@ class Surveyor:
                 self.summ_model.save_pretrained(models_dir + "/summ_model")
             #self.summ_tokenizer.save_pretrained(models_dir + "/summ_tokenizer")
             self.model = Summarizer(custom_model=self.summ_model, custom_tokenizer=self.summ_tokenizer)
-
-            if 't5' not in ledmodel_name:
-                self.ledtokenizer = LEDTokenizer.from_pretrained(ledmodel_name)
-                self.ledmodel = LEDForConditionalGeneration.from_pretrained(ledmodel_name).to(self.torch_device)
-            else:
-                self.ledtokenizer = AutoTokenizer.from_pretrained(ledmodel_name)
-                self.ledmodel = T5ForConditionalGeneration.from_pretrained(ledmodel_name).to(self.torch_device)
-
+            
             if 'led' in ledmodel_name:
                 self.ledtokenizer = LEDTokenizer.from_pretrained(ledmodel_name)
                 self.ledmodel = LEDForConditionalGeneration.from_pretrained(ledmodel_name).to(self.torch_device)