Update app.py

app.py

@@ -513,7 +513,7 @@ logging.basicConfig(level=logging.INFO, filename='translation.log', filemode='a'
                             format='%(asctime)s - %(levelname)s - %(message)s')
 
 # Main function to handle the translation workflow
-def main(model_id, dataset_url, model_type, output_dataset_name, range_specification):
+def main(dataset_url, model_type, output_dataset_name, range_specification):
     try:
         # Login to Hugging Face
         token = login()
@@ -586,11 +586,10 @@ with gr.Blocks() as demo:
 
     with gr.Row(equal_height=False):
         with gr.Column():
-            model_id = gr.Textbox(label="Model ID or URL", lines=1)
             dataset_url = gr.Textbox(label="Dataset URL", lines=1)
             model_type = gr.Dropdown(choices=["mix", "orpo", "ufb"], label="Model 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, 200-250")
+            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():
@@ -598,6 +597,6 @@ with gr.Blocks() as demo:
             logout_button = gr.Button("Logout")
 
     submit_btn = gr.Button("Translate Dataset", variant="primary")
-    submit_btn.click(main, inputs=[model_id, 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], outputs=output)
 
 demo.launch()