Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -11,8 +11,8 @@ def load_models():
|
|
11 |
model_name_dict = {
|
12 |
#'nllb-distilled-600M': 'facebook/nllb-200-distilled-600M',
|
13 |
#'nllb-1.3B': 'facebook/nllb-200-1.3B',
|
14 |
-
|
15 |
-
'nllb-3.3B': 'facebook/nllb-200-3.3B',
|
16 |
}
|
17 |
|
18 |
model_dict = {}
|
@@ -29,7 +29,7 @@ def load_models():
|
|
29 |
|
30 |
def translation(source, target, text):
|
31 |
if len(model_dict) == 2:
|
32 |
-
model_name = 'nllb-distilled-
|
33 |
|
34 |
start_time = time.time()
|
35 |
source = flores_codes[source]
|
|
|
11 |
model_name_dict = {
|
12 |
#'nllb-distilled-600M': 'facebook/nllb-200-distilled-600M',
|
13 |
#'nllb-1.3B': 'facebook/nllb-200-1.3B',
|
14 |
+
'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B',
|
15 |
+
#'nllb-3.3B': 'facebook/nllb-200-3.3B',
|
16 |
}
|
17 |
|
18 |
model_dict = {}
|
|
|
29 |
|
30 |
def translation(source, target, text):
|
31 |
if len(model_dict) == 2:
|
32 |
+
model_name = 'nllb-distilled-1.3B'
|
33 |
|
34 |
start_time = time.time()
|
35 |
source = flores_codes[source]
|