Update app.py

app.py

@@ -32,18 +32,18 @@ def translate(text, source_lang, target_lang, max_length=400):
     """
     Translate text from source language to target language
     """
-    src_lang = choose_language(source_lang)
-    tgt_lang= choose_language(target_lang)
-    if src_lang==None:
-        return "Error: the source langage is incorrect"
-    elif tgt_lang==None:
-        return "Error: the target language is incorrect"
+    # src_lang = choose_language(source_lang)
+    # tgt_lang= choose_language(target_lang)
+    # if src_lang==None:
+    #     return "Error: the source langage is incorrect"
+    # elif tgt_lang==None:
+    #     return "Error: the target language is incorrect"
 
     translation_pipeline = pipeline(TASK,
                                     model=model,
                                     tokenizer=tokenizer,
-                                    src_lang=src_lang,
-                                    tgt_lang=tgt_lang,
+                                    src_lang=source_lang,
+                                    tgt_lang=target_lang,
                                     max_length=max_length,
                                     device=device)
     result = translation_pipeline(text)