TiberiuCristianLeon commited on
Commit
007e517
·
verified ·
1 Parent(s): 579b2fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -20,7 +20,10 @@ st.session_state["tselected_language"] = tselected_language
20
  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(model_name)
25
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
26
  else:
 
20
  st.session_state["model_name"] = model_name
21
 
22
  if model_name == 'Helsinki-NLP':
23
+ try:
24
+ model_name = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
25
+ except EnvironmentError:
26
+ model_name = f"Helsinki-NLP/opus-tatoeba-{sl}-{tl}"
27
  tokenizer = AutoTokenizer.from_pretrained(model_name)
28
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
29
  else: