nouman66's picture
Update app.py
ec09ed1
raw
history blame
No virus
1.04 kB
import streamlit as st
from transformers import pipeline
st.title("Multilingual Translator Chatbot")
# Choose the translation model from Hugging Face
translation_model = st.selectbox("Select translation model", ["Helsinki-NLP/opus-mt-en-de", "Helsinki-NLP/opus-mt-en-fr"])
# Load the translation pipeline
translator = pipeline(task="translation", model=translation_model)
# User input for translation
user_input = st.text_area("Enter text for translation:", "")
if st.button("Translate"):
if user_input:
# Perform translation
translated_text = translator(user_input, max_length=500)[0]['translation_text']
st.success(f"Translated Text: {translated_text}")
else:
st.warning("Please enter text for translation.")
st.markdown("---")
st.subheader("About")
st.write(
"This is a simple Multilingual Translator chatbot that uses the Hugging Face Transformers library."
)
st.write(
"Select a translation model from the dropdown, enter text, and click 'Translate' to see the translation."
)