Update app.py
Browse files
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)
|