clean up

app.py

@@ -39,22 +39,21 @@ def identify_language(text):
 
 
 def translate(text, src_lang, tgt_lang):
-    src_lang_code = language_code_map[src_lang]
-    tgt_lang_code = language_code_map[tgt_lang]
-
     translation_pipeline = pipeline(
-        "translation", model=model, tokenizer=tokenizer, src_lang=src_lang_code, tgt_lang=tgt_lang_code, device=device)
+        "translation", model=model, tokenizer=tokenizer, src_lang=src_lang, tgt_lang=tgt_lang, device=device)
     result = translation_pipeline(text)
     return result[0]['translation_text']
 
 
 def query(text, user_lang):
-    detected_lang = identify_language(text)
+    detected_lang_code = identify_language(text)
+
+    user_lang_code = language_code_map[user_lang]
 
     translation = translate(
-        text, src_lang, tgt_lang) if detected_lang != user_lang else "User's content language is the same as the language of the input"
+        text, detected_lang_code, user_lang_code) if detected_lang_code != user_lang_code else "N/A \n(User's content language is the same as the language of the input)"
 
-    return [detected_lang, translation]
+    return [detected_lang_code, translation]
 
 
 examples = [
@@ -69,7 +68,7 @@ gr.Interface(
     [
         gr.Textbox(lines=3, label="User Input"),
         gr.Radio(["English", "Spanish", "Korean", "French", "German", "Japanese"],
-                 value="English", label="User's Content Language"),
+                 value="English", label="User Content Language Settings"),
     ],
     outputs=[
         gr.Textbox(lines=1, label="Detected Language"),