zionia commited on
Commit
353fb7a
1 Parent(s): 7335d74

revert to original

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="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,7 +22,7 @@ def translate(inp, direction):
22
  description = """
23
  <p>
24
  <center>
25
- Multi-domain Translation Between Siswati and English
26
  </center>
27
  </p>
28
  """
 
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
  description = """
23
  <p>
24
  <center>
25
+ Multi-domain Translation Between English and Vietnamese
26
  </center>
27
  </p>
28
  """