Update src/paraphrase/Paraphrase.py

src/paraphrase/Paraphrase.py

@@ -5,8 +5,8 @@ import src.exception.Exception.Exception as ExceptionCustom
 
 METHOD = "PARAPHRASE"
 
-tokenizer = AutoTokenizer.from_pretrained("BlackKakapo/flan-t5-base-paraphrase-ro")
-model = AutoModelForSeq2SeqLM.from_pretrained("BlackKakapo/flan-t5-base-paraphrase-ro")
+tokenizer = AutoTokenizer.from_pretrained("t5-base")
+model = AutoModelForSeq2SeqLM.from_pretrained("t5-base")
 
 # device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # model.to(device)