fix inf token ui

app_text_classification.py

@@ -4,7 +4,6 @@ from io_utils import (
     read_scanners,
     write_scanners,
     read_inference_type,
-    write_inference_type,
     get_logs_file,
 )
 from wordings import INTRODUCTION_MD, CONFIRM_MAPPING_DETAILS_MD
@@ -14,6 +13,8 @@ from text_classification_ui_helpers import (
     check_dataset_and_get_split,
     check_model_and_show_prediction,
     write_column_mapping_to_config,
+    select_run_mode,
+    deselect_run_inference,
 )
 
 MAX_LABELS = 20
@@ -70,7 +71,7 @@ def get_demo(demo):
         run_local = gr.Checkbox(value=True, label="Run in this Space")
         use_inference = read_inference_type(uid) == "hf_inference_api"
         run_inference = gr.Checkbox(value=use_inference, label="Run with Inference API")
-        inference_token = gr.Textbox(value="", label="HF Token for Inference API", visible=False)
+        inference_token = gr.Textbox(value="", label="HF Token for Inference API", visible=False, interactive=True)
 
     with gr.Accordion(label="Scanner Advance Config (optional)", open=False):
         selected = read_scanners(uid)
@@ -106,8 +107,9 @@ def get_demo(demo):
 
     scanners.change(write_scanners, inputs=[scanners, uid_label])
 
-    run_inference.change(write_inference_type, inputs=[run_inference, uid_label], outputs=[inference_token])
+    run_inference.change(select_run_mode, inputs=[run_inference, uid_label], outputs=[inference_token, run_local])
 
+    run_local.change(deselect_run_inference, inputs=[run_local], outputs=[inference_token, run_inference])
     gr.on(
         triggers=[label.change for label in column_mappings],
         fn=write_column_mapping_to_config,

io_utils.py

@@ -64,8 +64,6 @@ def write_inference_type(use_inference, uid):
         # save inference_type to inference_type in yaml
         yaml.dump(config, f, Dumper=Dumper)
     f.close()
-    return (gr.update(visible=(use_inference == "hf_inference_api")))
-
 
 
 # read column mapping from yaml file

text_classification_ui_helpers.py

@@ -9,7 +9,8 @@ import gradio as gr
 from transformers.pipelines import TextClassificationPipeline
 
 from io_utils import (get_yaml_path, read_column_mapping, save_job_to_pipe,
-                      write_column_mapping, write_log_to_user_file)
+                      write_column_mapping, write_log_to_user_file,
+                      write_inference_type)
 from text_classification import (check_model, get_example_prediction,
                                  get_labels_and_features_from_dataset)
 from wordings import CONFIRM_MAPPING_DETAILS_FAIL_RAW, MAPPING_STYLED_ERROR_WARNING, CHECK_CONFIG_OR_SPLIT_RAW
@@ -41,7 +42,29 @@ def check_dataset_and_get_split(dataset_id, dataset_config):
         # gr.Warning(f"Failed to load dataset {dataset_id} with config {dataset_config}: {e}")
         pass
 
+def select_run_mode(run_inf, uid):
+    write_inference_type(run_inf, uid)
+    if run_inf:
+        return (
+            gr.update(visible=True),
+            gr.update(value=False))
+    else:
+        return (
+            gr.update(visible=False),
+            gr.update(value=True))
 
+def deselect_run_inference(run_local):
+    if run_local:
+        return (
+            gr.update(visible=False),
+            gr.update(value=False)
+        )
+    else:
+        return (
+            gr.update(visible=True),
+            gr.update(value=True)
+        )
+        
 def write_column_mapping_to_config(
     dataset_id, dataset_config, dataset_split, uid, *labels
 ):