fix app

app.py

@@ -4,7 +4,6 @@ import torch
 from sacremoses import MosesPunctNormalizer
 import re
 import unicodedata
-import sys
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -27,16 +26,8 @@ def fix_tokenizer(tokenizer, new_lang='ami_Latn'):
 
 fix_tokenizer(small_tokenizer)
 
-# Language mapping
-lang_map = {
-    "汉语 Chinese": "zho_Hant",
-    "Amis": "ami_Latn"
-}
-
 # Translation function
-def translate(text, src_lang_label, tgt_lang_label):
-    src_lang = lang_map[src_lang_label]
-    tgt_lang = lang_map[tgt_lang_label]
+def translate(text, src_lang, tgt_lang):
     tokenizer, model = small_tokenizer, small_model
     if src_lang == "zho_Hant":
         text = preproc_chinese(text)
@@ -69,23 +60,15 @@ def preproc_chinese(text):
     clean = replace_nonprint(clean)
     return unicodedata.normalize("NFKC", clean)
 
-# Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# AMIS - Chinese Translation Tool")
-    src_lang = gr.Radio(choices=["汉语 Chinese", "Amis"], value="汉语 Chinese", label="Source Language")
-    tgt_lang = gr.Radio(choices=["Amis", "汉语 Chinese"], value="Amis", label="Target Language")
+    src_lang = gr.Radio(choices=["zho_Hant", "ami_Latn"], value="zho_Hant", label="Source Language")
+    tgt_lang = gr.Radio(choices=["ami_Latn", "zho_Hant"], value="ami_Latn", label="Target Language")
     input_text = gr.Textbox(label="Input Text", placeholder="Enter text here...")
     output_text = gr.Textbox(label="Translated Text", interactive=False)
     translate_btn = gr.Button("Translate")
-    
-    # Link translation function
-    translate_btn.click(
-        translate, 
-        inputs=[input_text, src_lang, tgt_lang], 
-        outputs=output_text
-    )
-    
-    gr.Markdown("感謝您在此專案上的辛勤工作 這只是這些模型能力的一小部分展示。電子郵件: [email protected]")
+
+    translate_btn.click(translate, inputs=[input_text, src_lang, tgt_lang], outputs=output_text)
 
 if __name__ == "__main__":
     demo.launch()