Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,8 @@ from transformers import MarianMTModel, MarianTokenizer
|
|
4 |
# Function to load the translation model
|
5 |
@st.cache_resource
|
6 |
def load_model():
|
7 |
-
|
|
|
8 |
model = MarianMTModel.from_pretrained(model_name)
|
9 |
tokenizer = MarianTokenizer.from_pretrained(model_name)
|
10 |
return model, tokenizer
|
@@ -18,7 +19,7 @@ def translate_to_english(text, model, tokenizer):
|
|
18 |
|
19 |
# Streamlit UI setup
|
20 |
st.title("Language Translator to English")
|
21 |
-
st.write("Translate any language to English using Hugging Face models!")
|
22 |
|
23 |
# Text input field
|
24 |
input_text = st.text_area("Enter text in any language", "")
|
|
|
4 |
# Function to load the translation model
|
5 |
@st.cache_resource
|
6 |
def load_model():
|
7 |
+
# Use a model for translating to English (Supports various languages)
|
8 |
+
model_name = "Helsinki-NLP/opus-mt-en-ROMANCE"
|
9 |
model = MarianMTModel.from_pretrained(model_name)
|
10 |
tokenizer = MarianTokenizer.from_pretrained(model_name)
|
11 |
return model, tokenizer
|
|
|
19 |
|
20 |
# Streamlit UI setup
|
21 |
st.title("Language Translator to English")
|
22 |
+
st.write("Translate any language (Romanian, Spanish, French, etc.) to English using Hugging Face models!")
|
23 |
|
24 |
# Text input field
|
25 |
input_text = st.text_area("Enter text in any language", "")
|