TiberiuCristianLeon commited on
Commit
496c579
·
verified ·
1 Parent(s): 2045817

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,12 +16,12 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
16
  try:
17
  model_name_full = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
18
  tokenizer = AutoTokenizer.from_pretrained(model_name_full)
19
- model = AutoModelForSeq2SeqLM.from_pretrained(model_name_full, device_map="auto")
20
  except EnvironmentError:
21
  try :
22
  model_name_full = f"Helsinki-NLP/opus-tatoeba-{sl}-{tl}"
23
  tokenizer = AutoTokenizer.from_pretrained(model_name_full)
24
- model = AutoModelForSeq2SeqLM.from_pretrained(model_name_full, device_map="auto")
25
  except EnvironmentError as error:
26
  return f"Error finding model: {model_name_full}! Try other available language combination.", error
27
  elif model_name.startswith('facebook/nllb'):
 
16
  try:
17
  model_name_full = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
18
  tokenizer = AutoTokenizer.from_pretrained(model_name_full)
19
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name_full)
20
  except EnvironmentError:
21
  try :
22
  model_name_full = f"Helsinki-NLP/opus-tatoeba-{sl}-{tl}"
23
  tokenizer = AutoTokenizer.from_pretrained(model_name_full)
24
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name_full)
25
  except EnvironmentError as error:
26
  return f"Error finding model: {model_name_full}! Try other available language combination.", error
27
  elif model_name.startswith('facebook/nllb'):