DSatishchandra commited on
Commit
a16fa51
·
verified ·
1 Parent(s): b9e4244

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- model_name = "Helsinki-NLP/opus-mt-any-to-en"
 
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", "")