Spaces:
Running
Running
geonmo.gu
commited on
Commit
•
c14e153
1
Parent(s):
e707077
fix error
Browse files
app.py
CHANGED
|
@@ -27,6 +27,9 @@ def load_models():
|
|
| 27 |
|
| 28 |
|
| 29 |
def translation(source, target, text):
|
|
|
|
|
|
|
|
|
|
| 30 |
start_time = time.time()
|
| 31 |
source = flores_codes[source]
|
| 32 |
target = flores_codes[target]
|
|
|
|
| 27 |
|
| 28 |
|
| 29 |
def translation(source, target, text):
|
| 30 |
+
if len(model_dict) == 2:
|
| 31 |
+
model_name = 'nllb-distilled-600M'
|
| 32 |
+
|
| 33 |
start_time = time.time()
|
| 34 |
source = flores_codes[source]
|
| 35 |
target = flores_codes[target]
|