TiberiuCristianLeon commited on
Commit
cfebeae
·
verified ·
1 Parent(s): 3b3b983

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -75,7 +75,7 @@ if model_name == 'Helsinki-NLP':
75
  except EnvironmentError:
76
  model_name = f"Helsinki-NLP/opus-tatoeba-{sl}-{tl}"
77
  tokenizer = AutoTokenizer.from_pretrained(model_name)
78
- model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
79
  if model_name.startswith('t5'):
80
  tokenizer = T5Tokenizer.from_pretrained(model_name)
81
  model = T5ForConditionalGeneration.from_pretrained(model_name).to(device)
 
75
  except EnvironmentError:
76
  model_name = f"Helsinki-NLP/opus-tatoeba-{sl}-{tl}"
77
  tokenizer = AutoTokenizer.from_pretrained(model_name)
78
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name).to(device)
79
  if model_name.startswith('t5'):
80
  tokenizer = T5Tokenizer.from_pretrained(model_name)
81
  model = T5ForConditionalGeneration.from_pretrained(model_name).to(device)