File size: 1,127 Bytes
669738a 659981a 38651ca 659981a e4ce59a 659981a 01910b2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
import streamlit as st
from transformers import pipeline
def main():
st.title("Multilingual Translator")
# Get user input
input_text = st.text_area("Enter text to translate:")
# Select source and target languages
source_lang = st.selectbox("Select source language:", get_languages())
target_lang = st.selectbox("Select target language:", get_languages())
# Translate text
if st.button("Translate"):
if input_text:
translation = translate_text(input_text, source_lang, target_lang)
st.success(f"Translated text: {translation}")
else:
st.warning("Please enter text to translate.")
def get_languages():
# You can customize this list based on the languages you want to support
return ["en", "es", "fr", "de", "it", "ja", "ko", "zh-CN"]
def translate_text(text, source_lang, target_lang):
translator = pipeline(task="translation", model=f"{source_lang}-to-{target_lang}")
translation = translator(text, max_length=500)[0]['translation_text']
return translation
if __name__ == "__main__":
main()
streamlit run app.py
|