Update app.py

app.py

@@ -162,6 +162,9 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
         prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=False)
         outputs = pipe(prompt, max_new_tokens=256, do_sample=False)
         translated_text = outputs[0]["generated_text"]
+        start_marker = "<end_of_turn>"
+        if start_marker in translated_text:
+            translated_text = translated_text.split(start_marker)[1].strip()
         return translated_text, message_text
     
     if model_name.startswith('t5'):
@@ -195,7 +198,7 @@ def create_interface():
         with gr.Row():
             sselected_language = gr.Dropdown(choices=options, value = options[0], label="Source language", interactive=True)
             tselected_language = gr.Dropdown(choices=options, value = options[1], label="Target language", interactive=True)
-            swap_button = gr.Button("Swap Languages")
+            swap_button = gr.Button("Swap Languages", size="md")
             swap_button.click(fn=swap_languages, inputs=[sselected_language, tselected_language], outputs=[sselected_language, tselected_language])
 
         model_name = gr.Dropdown(choices=models, label="Select a model", value = models[4], interactive=True)