Mediocreatmybest commited on
Commit
0bafb6f
1 Parent(s): 37854cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -3
app.py CHANGED
@@ -12,8 +12,8 @@ translation_task_names = {
12
  # 'German to English': 'translation_de_to_en',
13
  # 'English to Italian': 'translation_en_to_it',
14
  # 'Italian to English': 'translation_it_to_en',
15
- # 'English to Dutch': 'translation_en_to_nl',
16
- # 'Dutch to English': 'translation_nl_to_en',
17
  # 'English to Portuguese': 'translation_en_to_pt',
18
  # 'Portuguese to English': 'translation_pt_to_en',
19
  # 'English to Russian': 'translation_en_to_ru',
@@ -23,13 +23,21 @@ translation_task_names = {
23
  # 'English to Japanese': 'translation_en_to_ja',
24
  # 'Japanese to English': 'translation_ja_to_en',
25
  'English to Romanian': 'translation_en_to_ro',
 
26
  }
27
 
28
  model_names = {
29
  'T5-Base': 't5-base',
30
  'T5-Small': 't5-small',
31
  'T5-Large': 't5-large',
32
- 'Opus-En-ZH': 'liam168/trans-opus-mt-en-zh'
 
 
 
 
 
 
 
33
  }
34
 
35
  # Create a dictionary to store loaded models
 
12
  # 'German to English': 'translation_de_to_en',
13
  # 'English to Italian': 'translation_en_to_it',
14
  # 'Italian to English': 'translation_it_to_en',
15
+ 'English to Dutch': 'translation_en_to_nl',
16
+ 'Dutch to English': 'translation_nl_to_en',
17
  # 'English to Portuguese': 'translation_en_to_pt',
18
  # 'Portuguese to English': 'translation_pt_to_en',
19
  # 'English to Russian': 'translation_en_to_ru',
 
23
  # 'English to Japanese': 'translation_en_to_ja',
24
  # 'Japanese to English': 'translation_ja_to_en',
25
  'English to Romanian': 'translation_en_to_ro',
26
+ 'Swedish to English': 'translation_SV_to_EN',
27
  }
28
 
29
  model_names = {
30
  'T5-Base': 't5-base',
31
  'T5-Small': 't5-small',
32
  'T5-Large': 't5-large',
33
+ 'Opus-En-ZH': 'liam168/trans-opus-mt-en-zh',
34
+ 'DDDSSS/translation_en-zh': 'DDDSSS/translation_en-zh',
35
+ 'T5-Base-nl-en': 'yhavinga/t5-base-36L-ccmatrix-multi',
36
+ 'T5-Small-nl-en': 'yhavinga/t5-small-24L-ccmatrix-multi',
37
+ 'Opus-Sv-En-Test': 'Helsinki-NLP/opus-mt-sv-en',
38
+
39
+
40
+
41
  }
42
 
43
  # Create a dictionary to store loaded models