Update app.py

app.py

@@ -70,12 +70,12 @@ 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)
-        model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+        # tokenizer = AutoTokenizer.from_pretrained(model_name)
+        # model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
     except EnvironmentError:
         model_name = f"Helsinki-NLP/opus-tatoeba-{sl}-{tl}"
-        tokenizer = AutoTokenizer.from_pretrained(model_name)
-        model = AutoModelForSeq2SeqLM.from_pretrained(model_name).to(device)
+        # tokenizer = AutoTokenizer.from_pretrained(model_name)
+        # model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 if model_name.startswith('t5'):
     tokenizer = T5Tokenizer.from_pretrained(model_name)
     model = T5ForConditionalGeneration.from_pretrained(model_name).to(device)
@@ -87,13 +87,18 @@ translated_textarea = st.text("")
 # Handle the submit button click
 if submit_button:
     if model_name.startswith('Helsinki-NLP'):
-        prompt = input_text
-        print(prompt)
-        input_ids = tokenizer.encode(prompt, return_tensors='pt')
-        # Perform translation
-        output_ids = model.generate(input_ids)
-        # Decode the translated text
-        translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
+        # prompt = input_text
+        # print(prompt)
+        # input_ids = tokenizer.encode(prompt, return_tensors='pt')
+        # # Perform translation
+        # output_ids = model.generate(input_ids)
+        # # Decode the translated text
+        # translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
+        # Use a pipeline as a high-level helper
+        pipe = pipeline("translation", model=model_name)
+        translation = pipe(input_text)
+        translated_text = translation[0]['translation_text']
+
     elif model_name.startswith('Google'): 
         url = os.environ['GCLIENT'] + f'sl={sl}&tl={tl}&q={input_text}'
         response = httpx.get(url)