Update src/paraphrase/Paraphrase.py

src/paraphrase/Paraphrase.py

@@ -1,3 +1,4 @@
+import nltk
 from nltk.tokenize import sent_tokenize
 from transformers import PegasusForConditionalGeneration, PegasusTokenizer
 import torch
@@ -6,6 +7,8 @@ import src.exception.Exception.Exception as ExceptionCustom
 METHOD = "PARAPHRASE"
 
 def paraphraseParaphraseMethod(requestValue : str, model: str = 'tuner007/pegasus_paraphrase'):
+    nltk.download('punkt')
+    nltk.download('punkt_tab')
     exception = ""
     result_value = ""
     tokenizer = PegasusTokenizer.from_pretrained(model)