update

utils/__init__.py

@@ -1,2 +1,2 @@
 from utils.preprocess import read_text_to_json, convert_to_sentence_json, extract_sentence_features, is_valid_format
-from utils.methods import load_ExtModel, load_AbstrModel
+from utils.methods import load_ExtModel, load_AbstrModel, extractive_method, abstractive_method

utils/methods.py

@@ -83,5 +83,16 @@ def load_AbstrModel(path, device='cpu'):
     tokenizer = AutoTokenizer.from_pretrained(model_checkpoint, model_max_length=1024)
     abstrModel = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
     abstrModel = abstrModel.to(device)
+
+    generation_config = {
+        'num_beams': 5,
+        'max_length': 512,
+        'min_length': 64,
+        'length_penalty': 2.0,
+        'early_stopping': True,
+        'no_repeat_ngram_size': 3
+    }
+
+    abstrModel.config.update(generation_config)
     return tokenizer, abstrModel