inoki-giskard commited on
Commit
1afc3d2
1 Parent(s): e95a647

Update scan config yaml file path

Browse files
Files changed (1) hide show
  1. text_classification_ui_helpers.py +2 -2
text_classification_ui_helpers.py CHANGED
@@ -8,7 +8,7 @@ import datasets
8
  import gradio as gr
9
  from transformers.pipelines import TextClassificationPipeline
10
 
11
- from io_utils import (read_column_mapping, save_job_to_pipe,
12
  write_column_mapping, write_log_to_user_file)
13
  from text_classification import (check_model, get_example_prediction,
14
  get_labels_and_features_from_dataset)
@@ -217,7 +217,7 @@ def try_submit(m_id, d_id, config, split, local, uid):
217
  "--label_mapping",
218
  json.dumps(label_mapping),
219
  "--scan_config",
220
- "../config.yaml",
221
  ]
222
 
223
  eval_str = f"[{m_id}]<{d_id}({config}, {split} set)>"
 
8
  import gradio as gr
9
  from transformers.pipelines import TextClassificationPipeline
10
 
11
+ from io_utils import (get_yaml_path, read_column_mapping, save_job_to_pipe,
12
  write_column_mapping, write_log_to_user_file)
13
  from text_classification import (check_model, get_example_prediction,
14
  get_labels_and_features_from_dataset)
 
217
  "--label_mapping",
218
  json.dumps(label_mapping),
219
  "--scan_config",
220
+ get_yaml_path(uid),
221
  ]
222
 
223
  eval_str = f"[{m_id}]<{d_id}({config}, {split} set)>"