|
14 | 14 |
|
15 | 15 | def dingo_demo( |
16 | 16 | uploaded_file, |
17 | | - dataset_source, data_format, save_data, input_path, max_workers, batch_size, |
| 17 | + dataset_source, data_format, remove_output, input_path, max_workers, batch_size, |
18 | 18 | fields_data, |
19 | 19 | rule_list, llm_list, |
20 | 20 | # rule_config_data, |
@@ -180,7 +180,7 @@ def dingo_demo( |
180 | 180 | for item in detail: |
181 | 181 | new_detail.append(item) |
182 | 182 | if summary['output_path']: |
183 | | - if save_data == "false": |
| 183 | + if remove_output == "true": |
184 | 184 | shutil.rmtree(summary['output_path']) |
185 | 185 | summary['output_path'] = "" |
186 | 186 |
|
@@ -357,10 +357,10 @@ def get_data_column_mapping(): |
357 | 357 | ["jsonl", "json", "plaintext", "listjson","image"], |
358 | 358 | label="data_format" |
359 | 359 | ) |
360 | | - save_data = gr.Dropdown( |
| 360 | + remove_output = gr.Dropdown( |
361 | 361 | ["true", "false"], |
362 | | - value="false", |
363 | | - label="save_data" |
| 362 | + value="true", |
| 363 | + label="remove_output" |
364 | 364 | ) |
365 | 365 | with gr.Row(): |
366 | 366 | max_workers = gr.Number( |
@@ -483,7 +483,7 @@ def get_data_column_mapping(): |
483 | 483 | fn=dingo_demo, |
484 | 484 | inputs=[ |
485 | 485 | uploaded_file, |
486 | | - dataset_source, data_format, save_data, input_path, max_workers, batch_size, |
| 486 | + dataset_source, data_format, remove_output, input_path, max_workers, batch_size, |
487 | 487 | fields_dataframe, |
488 | 488 | rule_list, llm_list, |
489 | 489 | # rule_config_dataframe, |
|
0 commit comments