Update app.py

app.py

@@ -63,17 +63,17 @@ def polish(sentence: str, choice: str) -> str:
         return ""
     if choice.startswith("FLAN"):
         return run_flan(sentence)
-    elif choice.startswith("Round-trip"):
+    elif choice.startswith("Helsinki"):
         return run_roundtrip(sentence)
     else:
         return "Unknown option."
 
 # --- Gradio UI ---
 with gr.Blocks(title="English Grammar Polisher") as demo:
-    gr.Markdown("### English Grammar Polisher\nChoose FLAN-T5 (Google) or OPUS-MT round-trip (Helsinki-NLP).")
-    inp = gr.Textbox(lines=3, label="Input (English)", placeholder="Type a sentence…")
-    choice = gr.Dropdown(choices=MODEL_OPTIONS, value="FLAN-T5-base (Google en→en)", label="Method")
-    btn = gr.Button("Polish")
+    gr.Markdown("### English Grammar Polisher\nChoose FLAN-T5 (Google) or OPUS-MT (Helsinki-NLP).")
+    inp = gr.Textbox(lines=3, label="Input (English)", placeholder="Type an English sentence to correct.")
+    choice = gr.Dropdown(choices=MODEL_OPTIONS, value="FLAN-T5-base (Google)", label="Method")
+    btn = gr.Button("Oxford grammar polish")
     out = gr.Textbox(label="Output")
     btn.click(polish, inputs=[inp, choice], outputs=out)