Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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("
|
26 |
-
st.write("
|
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")
|