File size: 1,806 Bytes
ca40be0
 
 
 
228b24d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ca40be0
 
 
 
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
st.write(f"**Translated text in {target_language}:**")
st.write(translated_text)
st.write(f"""**Translated text in {target_language}:**""")
st.write(translated_text)
import streamlit as st
from transformers import pipeline

# Load the translation pipeline from Hugging Face
@st.cache_resource
def load_translation_pipeline(model_name="Helsinki-NLP/opus-mt-en-ROMANCE"):
    return pipeline("translation", model=model_name)

# Function to translate text
def translate_text(translation_pipeline, text, target_language):
    # Modify the model name to reflect the target language
    if target_language == "French":
        translation_pipeline.model_name = "Helsinki-NLP/opus-mt-en-fr"
    elif target_language == "Spanish":
        translation_pipeline.model_name = "Helsinki-NLP/opus-mt-en-es"
    elif target_language == "German":
        translation_pipeline.model_name = "Helsinki-NLP/opus-mt-en-de"
    else:
        st.error("Target language not supported!")
        return None
    
    translation = translation_pipeline(text)
    return translation[0]['translation_text']

# Streamlit app layout
st.title("Language Translator")

# Input text to translate
text = st.text_area("Enter text in English to translate:")

# Select target language
target_language = st.selectbox(
    "Select target language:",
    ["French", "Spanish", "German"]  # Add more languages if needed
)

# Load the translation pipeline
translation_pipeline = load_translation_pipeline()

# Translate button
if st.button("Translate"):
    if text:
        translated_text = translate_text(translation_pipeline, text, target_language)
        if translated_text:
            st.write(f"**Translated text in {target_language}:**")
            st.write(translated_text)
    else:
        st.error("Please enter text to translate.")