Create traducao.py

pages/traducao.py

@@ -0,0 +1,23 @@
+#Function to return the response
+def load_answer(question):
+    tokenizer = AutoTokenizer.from_pretrained("unicamp-dl/translation-pt-en-t5")
+    model = AutoModelForSeq2SeqLM.from_pretrained("unicamp-dl/translation-pt-en-t5")
+    pten_pipeline = pipeline('text2text-generation', model=model, tokenizer=tokenizer)
+    return pten_pipeline(question)[0]["generated_text"]
+
+#Gets the user input
+def get_text():
+    input_text = st.text_input("Sua frase em português: ", key="input")
+    return input_text
+
+user_input=get_text()
+response = load_answer(user_input)
+
+submit = st.button('Traduzir para inglês')
+
+
+
+if submit:
+
+    st.subheader("Answer:")
+    st.write(response)