Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ def translate_du(text, src_lang, tgt_lang, CKPT, max_length=400):
|
|
43 |
tokenizer=tokenizer,
|
44 |
src_lang=src_lang,
|
45 |
tgt_lang=tgt_lang,
|
46 |
-
|
47 |
max_length=max_length,
|
48 |
device=device)
|
49 |
|
|
|
43 |
tokenizer=tokenizer,
|
44 |
src_lang=src_lang,
|
45 |
tgt_lang=tgt_lang,
|
46 |
+
model = du_model,
|
47 |
max_length=max_length,
|
48 |
device=device)
|
49 |
|