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", "")
|