Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,12 @@ models = {
|
|
12 |
"Russian": "Helsinki-NLP/opus-mt-en-ru",
|
13 |
"Chinese": "Helsinki-NLP/opus-mt-en-zh",
|
14 |
"Japanese": "Helsinki-NLP/opus-mt-en-ja",
|
15 |
-
"Korean": "Helsinki-NLP/opus-mt-en-ko"
|
|
|
|
|
|
|
|
|
|
|
16 |
}
|
17 |
|
18 |
def load_model_and_tokenizer(model_name):
|
@@ -23,7 +28,7 @@ def load_model_and_tokenizer(model_name):
|
|
23 |
def translate_text(text, target_lang):
|
24 |
model_name = models.get(target_lang, None)
|
25 |
if not model_name:
|
26 |
-
return "
|
27 |
|
28 |
try:
|
29 |
model, tokenizer = load_model_and_tokenizer(model_name)
|
@@ -35,7 +40,7 @@ def translate_text(text, target_lang):
|
|
35 |
return f"Error during translation: {str(e)}"
|
36 |
|
37 |
def main():
|
38 |
-
st.title("Language Translator")
|
39 |
|
40 |
# User input
|
41 |
input_text = st.text_area("Enter text in English:", "")
|
|
|
12 |
"Russian": "Helsinki-NLP/opus-mt-en-ru",
|
13 |
"Chinese": "Helsinki-NLP/opus-mt-en-zh",
|
14 |
"Japanese": "Helsinki-NLP/opus-mt-en-ja",
|
15 |
+
"Korean": "Helsinki-NLP/opus-mt-en-ko",
|
16 |
+
"Arabic": "Helsinki-NLP/opus-mt-en-ar",
|
17 |
+
"Hindi": "Helsinki-NLP/opus-mt-en-hi",
|
18 |
+
"Thai": "Helsinki-NLP/opus-mt-en-th",
|
19 |
+
"Vietnamese": "Helsinki-NLP/opus-mt-en-vi",
|
20 |
+
"Swahili": "Helsinki-NLP/opus-mt-en-sw"
|
21 |
}
|
22 |
|
23 |
def load_model_and_tokenizer(model_name):
|
|
|
28 |
def translate_text(text, target_lang):
|
29 |
model_name = models.get(target_lang, None)
|
30 |
if not model_name:
|
31 |
+
return "Translation model for the selected language is not available."
|
32 |
|
33 |
try:
|
34 |
model, tokenizer = load_model_and_tokenizer(model_name)
|
|
|
40 |
return f"Error during translation: {str(e)}"
|
41 |
|
42 |
def main():
|
43 |
+
st.title("Universal Language Translator")
|
44 |
|
45 |
# User input
|
46 |
input_text = st.text_area("Enter text in English:", "")
|