Update app.py

app.py

@@ -28,16 +28,12 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 
 
 @lru_cache(maxsize=100)
-def translate(text: str, src_lang: str, tgt_lang: str):
-    if not src_lang:
-        raise gr.Error("The source language is empty! Please choose it in the dropdown list.")
-    if not tgt_lang:
-        raise gr.Error("The target language is empty! Please choose it in the dropdown list.")
-    return _translate(text, src_lang, tgt_lang)
+def translate(text: str):
+    return _translate(text)
 
 # Only assign GPU if cache not used
 @spaces.GPU
-def _translate(text: str, src_lang: str, tgt_lang: str):
+def _translate(text: str):
     input_tokens = (
         tokenizer(text, return_tensors="pt")
         .input_ids[0]
@@ -88,7 +84,7 @@ with gr.Blocks() as demo:
         output = gr.Textbox(label="Output Text", lines=6)
     btn.click(
         translate,
-        inputs=[input_text, src_lang, target_lang],
+        inputs=[input_text],
         outputs=output,
     )
     examples = gr.Examples(examples=examples_inputs,inputs=[input_text, src_lang,target_lang], fn=translate, outputs=output, cache_examples=True)