Spaces:

vijay399
/

testing

Running

vijay399 commited on Apr 24, 2023

Commit

38b6bd6

1 Parent(s): 02dbb6b

Update src/paraphrase/Paraphrase.py

Files changed (1) hide show

src/paraphrase/Paraphrase.py CHANGED Viewed

@@ -5,8 +5,9 @@ import src.exception.Exception.Exception as ExceptionCustom
 METHOD = "PARAPHRASE"
-tokenizer = PegasusTokenizer.from_pretrained('google/pegasus-xsum')
-model = PegasusForConditionalGeneration.from_pretrained('google/pegasus-xsum')
 def paraphraseParaphraseMethod(requestValue : str):
     exception = ""

 METHOD = "PARAPHRASE"
+tokenizer = PegasusTokenizer.from_pretrained('tuner007/pegasus_paraphrase')
+model = PegasusForConditionalGeneration.from_pretrained('tuner007/pegasus_paraphrase')
 def paraphraseParaphraseMethod(requestValue : str):
     exception = ""