Hugging Face's logo

Spaces:
jonathanjordan21
/
mrebel-demo
Sleeping

App Files Community
PereLluis13 commited on
Commit
3b326f4
•
1 Parent(s): e6f2745

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ import torch
14
  )
15
  def load_models(lan):
16
  st_time = time()
17
- tokenizer = AutoTokenizer.from_pretrained("Babelscape/mrebel-large", src_lang=_Tokens[lan], "tgt_lang": "tp_XX")
18
  print("+++++ loading Model", time() - st_time)
19
  model = AutoModelForSeq2SeqLM.from_pretrained("Babelscape/mrebel-large")
20
  if torch.cuda.is_available():
 
14
  )
15
  def load_models(lan):
16
  st_time = time()
17
+ tokenizer = AutoTokenizer.from_pretrained("Babelscape/mrebel-large", src_lang=_Tokens[lan], tgt_lang="tp_XX")
18
  print("+++++ loading Model", time() - st_time)
19
  model = AutoModelForSeq2SeqLM.from_pretrained("Babelscape/mrebel-large")
20
  if torch.cuda.is_available():