Update model.py

model.py

@@ -18,7 +18,7 @@ from tqdm.notebook import tqdm
 class Abstractive_Summarization_Model:
     def __init__(self):
         self.text = None
-        self.IS_CNNDM = False # whether to use CNNDM dataset or XSum dataset
+        self.IS_CNNDM = True # whether to use CNNDM dataset or XSum dataset
         self.LOWER = False
         self.max_length = 1024 #if self.IS_CNNDM else 512
         self.model, self.tokenizer  = self.load_model()