Spaces:
Build error
Build error
Commit
·
b51f7b9
1
Parent(s):
2f54e99
additional commits
Browse files
app.py
CHANGED
|
@@ -49,10 +49,11 @@ def load_model(source_language_,target_language_):
|
|
| 49 |
|
| 50 |
params = load_params_for_prediction(parsed_yaml_file,best_ckpt)
|
| 51 |
return params
|
| 52 |
-
|
| 53 |
-
#Using global params for English and Swahili to save time taken to load them
|
|
|
|
| 54 |
params = load_model('English','Swahili')
|
| 55 |
-
|
| 56 |
def get_translation(source_language,target_language,source_sentence=None,source_file=None):
|
| 57 |
'''
|
| 58 |
This takes a sentence and gets the translation.
|
|
@@ -64,7 +65,8 @@ def get_translation(source_language,target_language,source_sentence=None,source_
|
|
| 64 |
type_=1
|
| 65 |
source = source_file.name
|
| 66 |
try:
|
| 67 |
-
|
|
|
|
| 68 |
|
| 69 |
pred = translate(params,source,type_)
|
| 70 |
except Exception:
|
|
|
|
| 49 |
|
| 50 |
params = load_params_for_prediction(parsed_yaml_file,best_ckpt)
|
| 51 |
return params
|
| 52 |
+
'''
|
| 53 |
+
#Using global params for English and Swahili (default pairs for this demo) to save time taken to load them.
|
| 54 |
+
params=None
|
| 55 |
params = load_model('English','Swahili')
|
| 56 |
+
'''
|
| 57 |
def get_translation(source_language,target_language,source_sentence=None,source_file=None):
|
| 58 |
'''
|
| 59 |
This takes a sentence and gets the translation.
|
|
|
|
| 65 |
type_=1
|
| 66 |
source = source_file.name
|
| 67 |
try:
|
| 68 |
+
|
| 69 |
+
params = load_model(source_language,target_language) #Uncomment when not Using global params for English and Swahili to save time taken to load them
|
| 70 |
|
| 71 |
pred = translate(params,source,type_)
|
| 72 |
except Exception:
|