Unityraptor commited on
Commit
7ab5037
1 Parent(s): 74217d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,7 +9,8 @@ tokenizer = AutoTokenizer.from_pretrained("humarin/chatgpt_paraphraser_on_T5_bas
9
 
10
  model = AutoModelForSeq2SeqLM.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base").to(device)
11
 
12
- translator = pipeline("translation", model="facebook/nllb-200-distilled-600M")
 
13
 
14
  def paraphrase(
15
  question,
 
9
 
10
  model = AutoModelForSeq2SeqLM.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base").to(device)
11
 
12
+ #translator = pipeline("translation", model="facebook/nllb-200-distilled-600M")
13
+ translator = pipeline("translation", model="google-t5/t5-large")
14
 
15
  def paraphrase(
16
  question,