import streamlit as st | |
from transformers import pipeline | |
translator = pipeline("translation", model="Helsinki-NLP/opus-mt-mul-en") | |
def main(): | |
st.title("translation") | |
with st.form("translator"): | |
text = st.text_area('enter some text:') | |
# clicked==True only when the button is clicked | |
clicked = st.form_submit_button("Submit") | |
if clicked: | |
results = translator([text]) | |
st.json(results) | |
if __name__ == "__main__": | |
main() |