Spaces:
Runtime error
Runtime error
Commit
•
0126e4b
1
Parent(s):
1a0e861
Update app.py
Browse files
app.py
CHANGED
@@ -7,14 +7,14 @@ from transformers import pipeline
|
|
7 |
|
8 |
def translate(text,number):
|
9 |
if number==1:
|
10 |
-
|
11 |
elif number==2:
|
12 |
-
|
13 |
else:
|
14 |
-
|
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 |
|