edaiofficial commited on
Commit
5e78dce
·
1 Parent(s): 7da89d6

additional commits

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -26,9 +26,12 @@ def load_config(path="configs/default.yaml") -> dict:
26
  cfg = yaml.safe_load(ymlfile)
27
  return cfg
28
 
29
- def load_model(source_language,target_language):
30
- source_language = 'en'
31
- target_language = 'sw'
 
 
 
32
  translation_dir = 'main'
33
 
34
  try:
 
26
  cfg = yaml.safe_load(ymlfile)
27
  return cfg
28
 
29
+ def load_model(source_language_,target_language_):
30
+ source_language = language_map[source_language_]
31
+ target_language = language_map[target_language_]
32
+
33
+ #source_language = 'en'
34
+ #target_language = 'sw'
35
  translation_dir = 'main'
36
 
37
  try: