RedDev commited on
Commit
8c3c25e
1 Parent(s): 1d40d85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
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())