Update app.py

app.py

@@ -12,10 +12,14 @@ tokenizer = AutoTokenizer.from_pretrained(CKPT)
 device = 0 if torch.cuda.is_available() else -1
 
 
-def translate(text, src_lang, tgt_lang, max_length=400):
+def translate(text):
     """
     Translate the text from source lang to target lang
     """
+    src_lang = "zho-Hans"
+    tgt_lang = "eng_Latn"
+    max_length = 400
+    
     translation_pipeline = pipeline(TASK,
                                     model=model,
                                     tokenizer=tokenizer,
@@ -32,9 +36,6 @@ gr.Interface(
     translate,
     [
         gr.components.Textbox(label="Text"),
-        gr.components.Dropdown(label="Source Language", choices=LANGS),
-        gr.components.Dropdown(label="Target Language", choices=LANGS),
-        gr.components.Slider(8, 512, value=400, step=8, label="Max Length")
     ],
     ["text"],
 ).launch()