Spaces:

Unityraptor
/

MediaRephraser

Sleeping

Unityraptor commited on Jun 3

Commit

f306387

•

1 Parent(s): 86c4485

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ tokenizer = AutoTokenizer.from_pretrained("humarin/chatgpt_paraphraser_on_T5_bas
 model = AutoModelForSeq2SeqLM.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base").to(device)
-translator = pipeline("translation", model="google-t5/t5-small")
 def paraphrase(
     question,

 model = AutoModelForSeq2SeqLM.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base").to(device)
+translator = pipeline("translation", model="facebook/nllb-200-distilled-600M")
 def paraphrase(
     question,