Update app.py

app.py

@@ -70,14 +70,14 @@ st.session_state["model_name"] = model_name
 if model_name == 'Helsinki-NLP':
     try:
         model_name = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
-        # tokenizer = AutoTokenizer.from_pretrained(model_name)
-        pipe = pipeline("translation", model=model_name)
-        # model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+        tokenizer = AutoTokenizer.from_pretrained(model_name)
+        model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+        pipe = pipeline("translation", model=model, tokenizer=tokenizer)
     except (EnvironmentError, OSError):
         model_name = f"Helsinki-NLP/opus-tatoeba-{sl}-{tl}"
-        # tokenizer = AutoTokenizer.from_pretrained(model_name)
-        pipe = pipeline("translation", model=model_name)
-        # model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+        tokenizer = AutoTokenizer.from_pretrained(model_name)
+        model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+        pipe = pipeline("translation", model=model, tokenizer=tokenizer)
 if model_name.startswith('t5'):
     tokenizer = T5Tokenizer.from_pretrained(model_name)
     model = T5ForConditionalGeneration.from_pretrained(model_name).to(device)