Update app.py

app.py

@@ -77,20 +77,18 @@ def translate_text(text, source_lang, target_lang):
             num_return_sequences=1,
         )
         full_output = tokenizer.decode(translated_chunk[0], skip_special_tokens=True)
-        full_output = full_output.replace(prompt.strip(),"")
+        full_output = full_output.replace(text.strip(),"")
         yield full_output
     except Exception as e:
         yield f"翻译出错: {str(e)}"
 
 
-description = """
-#👋 Seed-X, powered by Bytedance
-We are excited to introduce Seed-X, a powerful series of open-source multilingual translation language models, including an instruction model, a reinforcement learning model, and a reward model. It pushes the boundaries of translation capabilities within 7 billion parameters. We develop Seed-X as an accessible, off-the-shelf tool to support the community in advancing translation research and applications.
-"""
-
 # 创建 Gradio 界面
 with gr.Blocks(title="Seed-X") as demo:
-    gr.Markdown(description)
+    gr.Markdown("# 👋 Seed-X, powered by Bytedance")
+    gr.Markdown(
+        'A real-time translation tool based on Seed-X. It pushes the boundaries of translation capabilities within 7 billion parameters.'
+    )
 
     with gr.Column():
         with gr.Row():
@@ -104,6 +102,7 @@ with gr.Blocks(title="Seed-X") as demo:
                     value="English",
                     label="Target Language"
             )
+        with gr.Row():
             translate_btn = gr.Button("Translate", variant="primary")
         with gr.Row():
             source_text = gr.Textbox(