TANVEERMAKHDOOM commited on
Commit
86e2526
·
verified ·
1 Parent(s): 2c18d29

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
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 "Model for the selected language is not available."
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:", "")