minor uptdates

app.py

@@ -8,7 +8,7 @@ MODEL = MT5ForConditionalGeneration.from_pretrained("werent4/mt5TranslatorLT")
 MODEL.to(DEVICE)
 
 def translate(text, mode, max_length, num_beams):
-    text = f"translate English to Lithuanian: {text}" if mode == "En2Lt" else f"translate Lithuanian to English: {text}"
+    text = f"translate English to Lithuanian: {text}"
     encoded_input = TOKENIZER(text, return_tensors="pt", padding=True, truncation=True, max_length=max_length).to(DEVICE)
     with torch.no_grad():
         output_tokens = MODEL.generate(
@@ -30,7 +30,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=["En2Lt", "Lt2En"])
+      mode = gr.Dropdown(label="Mode", choices=["Currenly wroking only in en-lt", "Currenly wroking only in en-lt"])
       translate_button = gr.Button("Translate")
   output_text = gr.Textbox(label="Translated text")
   with gr.Accordion("How to run the model locally:", open=False):