Lenylvt commited on
Commit
2ff9784
Β·
verified Β·
1 Parent(s): 4c8613b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -22,17 +22,17 @@ url = "https://huggingface.co/Lenylvt/LanguageISO/resolve/main/iso.md"
22
  language_options = fetch_languages(url)
23
 
24
  # Streamlit UI components
25
- st.title("Text Translator with Dynamic Language Options")
26
- st.write("Select source and target languages to translate text.")
27
 
28
- source_language = st.selectbox("Select Source Language", options=language_options, format_func=lambda x: x[1])
29
- target_language = st.selectbox("Select Target Language", options=language_options, format_func=lambda x: x[1])
30
- text = st.text_area("Enter text to translate...", height=150)
31
 
32
  def translate_text(text, source_language_code, target_language_code):
33
  model_name = f"Helsinki-NLP/opus-mt-{source_language_code}-{target_language_code}"
34
  if source_language_code == target_language_code:
35
- return "Translation between the same languages is not supported."
36
  try:
37
  tokenizer = MarianTokenizer.from_pretrained(model_name)
38
  model = MarianMTModel.from_pretrained(model_name)
@@ -42,8 +42,8 @@ def translate_text(text, source_language_code, target_language_code):
42
  except Exception as e:
43
  return f"Failed to load model for {source_language_code} to {target_language_code}: {str(e)}"
44
 
45
- if st.button("Translate"):
46
  source_language_code, _ = source_language
47
  target_language_code, _ = target_language
48
  translation = translate_text(text, source_language_code, target_language_code)
49
- st.text_area("Translated Text", value=translation, height=150, key="translation_output")
 
22
  language_options = fetch_languages(url)
23
 
24
  # Streamlit UI components
25
+ st.title("πŸ“œ Translator")
26
+ st.write("We use model from [Language Technology Research Group at the University of Helsinki](https://huggingface.co/Helsinki-NLP). For API use please visit [this space](https://huggingface.co/spaces/Lenylvt/Translator-API)")
27
 
28
+ source_language = st.selectbox("1️⃣ Select Source Language", options=language_options, format_func=lambda x: x[1])
29
+ target_language = st.selectbox("2️⃣ Select Target Language", options=language_options, format_func=lambda x: x[1])
30
+ text = st.text_area("βœ’οΈ Enter text to translate...", height=150)
31
 
32
  def translate_text(text, source_language_code, target_language_code):
33
  model_name = f"Helsinki-NLP/opus-mt-{source_language_code}-{target_language_code}"
34
  if source_language_code == target_language_code:
35
+ return "πŸ”΄ Translation between the same languages is not supported."
36
  try:
37
  tokenizer = MarianTokenizer.from_pretrained(model_name)
38
  model = MarianMTModel.from_pretrained(model_name)
 
42
  except Exception as e:
43
  return f"Failed to load model for {source_language_code} to {target_language_code}: {str(e)}"
44
 
45
+ if st.button("πŸ“ Translate"):
46
  source_language_code, _ = source_language
47
  target_language_code, _ = target_language
48
  translation = translate_text(text, source_language_code, target_language_code)
49
+ st.text_area("⬇️ Translated Text", value=translation, height=150, key="translation_output")