Update app.py

app.py

@@ -576,7 +576,7 @@ gradio_desc = """This tool translates datasets using the WMT21 translation model
 ## 💭 What Does This Tool Do:
 - Translates datasets based on the selected model type.
 - Uploads the translated dataset to Hugging Face.
-## 📊 Dataset Model Types:
+## 📊 Dataset Types:
 - **mix**: 
   - `prompt`: List of dictionaries with 'content' and 'role' fields (multi-turn conversation).
   - `chosen`: Single dictionary with 'content' and 'role' fields.
@@ -592,24 +592,32 @@ gradio_desc = """This tool translates datasets using the WMT21 translation model
 ## 🛠️ Backend:
 The translation backend runs on the Hugging Face Hub API.
 """
+# Define the theme
+theme = gr.themes.Soft(text_size="lg", spacing_size="lg")
 
-with gr.Blocks() as demo:
+with gr.Blocks(theme=theme) as demo:
     gr.HTML(f"""<h1 align="center" id="space-title">{gradio_title}</h1>""")
     gr.Markdown(gradio_desc)
 
+    with gr.Row(variant="panel"):
+        gr.Markdown(value="## 🚀 Login to Hugging Face"),
+        token_input = gr.LoginButton(min_width=380)
+
+    gr.Markdown(value="🚨 **This is needed to upload the resulting dataset.**")
+
     with gr.Row(equal_height=False):
         with gr.Column():
-            dataset_url = gr.Textbox(label="Dataset URL", lines=1)
-            model_type = gr.Dropdown(choices=["mix", "orpo", "ufb"], label="Model Type")
+            dataset_url = gr.Textbox(label="Input Dataset URL", lines=1)
+            model_type = gr.Dropdown(choices=["mix", "orpo", "ufb"], label="Dataset Type")
             output_dataset_name = gr.Textbox(label="Output Dataset Name", lines=1)
             range_specification = gr.Textbox(label="Range Specification", lines=1, placeholder="e.g., 1-100")
-            login_button = gr.Button("Login to Hugging Face")
 
         with gr.Column():
             output = gr.Textbox(label="Output", lines=1)
-            logout_button = gr.Button("Logout")
 
     submit_btn = gr.Button("Translate Dataset", variant="primary")
-    submit_btn.click(main, inputs=[dataset_url, model_type, output_dataset_name, range_specification], outputs=output)
+    submit_btn.click(main, inputs=[dataset_url, model_type, output_dataset_name, range_specification, token_input], outputs=output)
+
+    gr.Markdown(value="We sincerely thank our community members for their extraordinary contributions to this project.")
 
-demo.launch()
+demo.launch(share=True)