zionia commited on
Commit
7335d74
·
verified ·
1 Parent(s): fc9b454

update model to ss-en-m2m100-combo

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -3,14 +3,14 @@ from gradio.mix import Parallel, Series
3
 
4
  from transformers import pipeline
5
 
6
- translater = pipeline("translation", model="VietAI/envit5-translation")
7
 
8
 
9
  def translate(inp, direction):
10
- if direction == 'en->vi':
11
- text = "en: " + inp
12
  else:
13
- text = "vi: " + inp
14
 
15
  res = translater(
16
  text,
@@ -22,7 +22,7 @@ def translate(inp, direction):
22
  description = """
23
  <p>
24
  <center>
25
- Multi-domain Translation Between English and Vietnamese
26
  </center>
27
  </p>
28
  """
 
3
 
4
  from transformers import pipeline
5
 
6
+ translater = pipeline("translation", model="dsfsi/ss-en-m2m100-combo")
7
 
8
 
9
  def translate(inp, direction):
10
+ if direction == 'si->en':
11
+ text = "si: " + inp
12
  else:
13
+ text = "en: " + inp
14
 
15
  res = translater(
16
  text,
 
22
  description = """
23
  <p>
24
  <center>
25
+ Multi-domain Translation Between Siswati and English
26
  </center>
27
  </p>
28
  """