TiberiuCristianLeon commited on
Commit
8a23bd0
·
verified ·
1 Parent(s): 582c23b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,8 +21,8 @@ st.session_state["model_name"] = model_name
21
 
22
  if model_name == 'Helsinki-NLP':
23
  model_name = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
24
- tokenizer = AutoTokenizer.from_pretrained(f"Helsinki-NLP/opus-mt-{sl}-{tl}")
25
- model = AutoModelForSeq2SeqLM.from_pretrained(f"Helsinki-NLP/opus-mt-{sl}-{tl}")
26
  else:
27
  tokenizer = T5Tokenizer.from_pretrained(model_name)
28
  model = T5ForConditionalGeneration.from_pretrained(model_name)
 
21
 
22
  if model_name == 'Helsinki-NLP':
23
  model_name = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
24
+ tokenizer = AutoTokenizer.from_pretrained(model_name")
25
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
26
  else:
27
  tokenizer = T5Tokenizer.from_pretrained(model_name)
28
  model = T5ForConditionalGeneration.from_pretrained(model_name)