Skip to content

Commit 4fa75c3

Browse files
committed
feat: remove_output
1 parent b4ae906 commit 4fa75c3

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

app_gradio/app.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414

1515
def dingo_demo(
1616
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,
1818
fields_data,
1919
rule_list, llm_list,
2020
# rule_config_data,
@@ -180,7 +180,7 @@ def dingo_demo(
180180
for item in detail:
181181
new_detail.append(item)
182182
if summary['output_path']:
183-
if save_data == "false":
183+
if remove_output == "true":
184184
shutil.rmtree(summary['output_path'])
185185
summary['output_path'] = ""
186186

@@ -357,10 +357,10 @@ def get_data_column_mapping():
357357
["jsonl", "json", "plaintext", "listjson","image"],
358358
label="data_format"
359359
)
360-
save_data = gr.Dropdown(
360+
remove_output = gr.Dropdown(
361361
["true", "false"],
362-
value="false",
363-
label="save_data"
362+
value="true",
363+
label="remove_output"
364364
)
365365
with gr.Row():
366366
max_workers = gr.Number(
@@ -483,7 +483,7 @@ def get_data_column_mapping():
483483
fn=dingo_demo,
484484
inputs=[
485485
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,
487487
fields_dataframe,
488488
rule_list, llm_list,
489489
# rule_config_dataframe,

0 commit comments

Comments
 (0)