Update src/paraphrase/Paraphrase.py

src/paraphrase/Paraphrase.py

@@ -5,13 +5,11 @@ import src.exception.Exception.Exception as ExceptionCustom
 
 METHOD = "PARAPHRASE"
 
-tokenizer = PegasusTokenizer.from_pretrained('tuner007/pegasus_paraphrase')
-model = PegasusForConditionalGeneration.from_pretrained('tuner007/pegasus_paraphrase')
-
-
-def paraphraseParaphraseMethod(requestValue : str, model: str):
+def paraphraseParaphraseMethod(requestValue : str, model: str = 'tuner007/pegasus_paraphrase'):
     exception = ""
     result_value = ""
+    tokenizer = PegasusTokenizer.from_pretrained(model)
+    model = PegasusForConditionalGeneration.from_pretrained(model)
 
     exception = ExceptionCustom.checkForException(requestValue, METHOD)
     if exception != "":