GiladtheFixer commited on
Commit
0126e4b
1 Parent(s): 1a0e861

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -7,14 +7,14 @@ from transformers import pipeline
7
 
8
  def translate(text,number):
9
  if number==1:
10
- model='Helsinki-NLP/opus-mt-en-es'
11
  elif number==2:
12
- model='Helsinki-NLP/opus-mt-en-fr'
13
  else:
14
- model='Helsinki-NLP/opus-mt-en-ru'
15
 
16
- pipe=pipeline(model)
17
- return pipe(text)
18
 
19
 
20
 
 
7
 
8
  def translate(text,number):
9
  if number==1:
10
+ model1='Helsinki-NLP/opus-mt-en-es'
11
  elif number==2:
12
+ model1='Helsinki-NLP/opus-mt-en-fr'
13
  else:
14
+ model1='Helsinki-NLP/opus-mt-en-ru'
15
 
16
+ pipe=pipeline("translation", model=model1)
17
+ return pipe(text)[0]["translation_text"]
18
 
19
 
20