Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -20,6 +20,7 @@ def fix_tokenizer(tokenizer, new_lang='tok_Latn'):
|
|
20 |
tokenizer.added_tokens_encoder = {}
|
21 |
tokenizer.added_tokens_decoder = {}
|
22 |
|
|
|
23 |
model_load_name = 'RedDev/nllb-deu-tok-v1'
|
24 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_load_name).cuda()
|
25 |
tokenizer = NllbTokenizer.from_pretrained(model_load_name)
|
@@ -46,10 +47,6 @@ LANG_CODES = {
|
|
46 |
|
47 |
if __name__ == '__main__':
|
48 |
print('\tinit models')
|
49 |
-
|
50 |
-
global model_dict
|
51 |
-
|
52 |
-
model_dict = load_models()
|
53 |
|
54 |
# define gradio demo
|
55 |
lang_codes = list(LANG_CODES.keys())
|
|
|
20 |
tokenizer.added_tokens_encoder = {}
|
21 |
tokenizer.added_tokens_decoder = {}
|
22 |
|
23 |
+
accelerator = Accelerator(cpu=True)
|
24 |
model_load_name = 'RedDev/nllb-deu-tok-v1'
|
25 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_load_name).cuda()
|
26 |
tokenizer = NllbTokenizer.from_pretrained(model_load_name)
|
|
|
47 |
|
48 |
if __name__ == '__main__':
|
49 |
print('\tinit models')
|
|
|
|
|
|
|
|
|
50 |
|
51 |
# define gradio demo
|
52 |
lang_codes = list(LANG_CODES.keys())
|