Unityraptor commited on
Commit
72c2b08
1 Parent(s): deb90aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -8,6 +8,8 @@ model = AutoModelForSeq2SeqLM.from_pretrained("humarin/chatgpt_paraphraser_on_T5
8
 
9
  translation = pipeline("translation", model="facebook/nllb-200-distilled-600M", src_lang='en', tgt_lang="fr")
10
 
 
 
11
  def paraphrase(
12
  question,
13
  num_beams=5,
@@ -39,7 +41,7 @@ def paraphrase(
39
 
40
 
41
  def summary(myinput):
42
- myout = translation(myinput, )
43
  return myout
44
 
45
  def launch(input):
 
8
 
9
  translation = pipeline("translation", model="facebook/nllb-200-distilled-600M", src_lang='en', tgt_lang="fr")
10
 
11
+ m2m100_en_fr = pipeline('translation', 'facebook/m2m100_418M', src_lang='en', tgt_lang="fr")
12
+
13
  def paraphrase(
14
  question,
15
  num_beams=5,
 
41
 
42
 
43
  def summary(myinput):
44
+ myout = m2m100_en_fr(myinput)
45
  return myout
46
 
47
  def launch(input):