space updated

app.py

@@ -7,7 +7,7 @@ TOKENIZER = T5Tokenizer.from_pretrained('werent4/mt5TranslatorLT')
 MODEL = MT5ForConditionalGeneration.from_pretrained("werent4/mt5TranslatorLT")
 MODEL.to(DEVICE)
 
-def translate(text, model, tokenizer, device, translation_way = "en-lt"):
+def translate(text, model,device, translation_way = "en-lt"):
     translations_ways = {
         "en-lt": "<EN2LT>",
         "lt-en": "<LT2EN>"
@@ -36,7 +36,7 @@ with gr.Blocks() as interface:
   with gr.Row():
     input_text = gr.Textbox(label="Text input", placeholder="Enter your text here")
     with gr.Column():
-      mode = gr.Dropdown(label="Mode", choices=["en-lt", "lt-en"])
+      translation_way = gr.Dropdown(label="Mode", choices=["en-lt", "lt-en"])
       translate_button = gr.Button("Translate")
   output_text = gr.Textbox(label="Translated text")
   with gr.Accordion("How to run the model locally:", open=False):