make inference api default; improve event triggers

app_text_classification.py

@@ -80,13 +80,13 @@ def get_demo():
                     for _ in range(MAX_LABELS, MAX_LABELS + MAX_FEATURES):
                         column_mappings.append(gr.Dropdown(visible=False))
 
-    with gr.Accordion(label="Model Wrap Advance Config (optional)", open=False):
-        run_local = gr.Checkbox(value=True, label="Run with Local Model Inference (pipeline)")
-        run_inference = gr.Checkbox(value=False, label="Run with Inference API")
+    with gr.Accordion(label="Model Wrap Advance Config", open=True):
+        run_local = gr.Checkbox(value=False, label="Run with Local Model Inference (pipeline)")
+        run_inference = gr.Checkbox(value=True, label="Run with Inference API")
         inference_token = gr.Textbox(
             value="",
             label="HF Token for Inference API",
-            visible=False,
+            visible=True,
             interactive=True,
         )
 
@@ -206,6 +206,8 @@ def get_demo():
             dataset_config_input,
             dataset_split_input,
             uid_label,
+            run_inference,
+            inference_token,
         ],
         outputs=[
             example_input,
@@ -235,7 +237,11 @@ def get_demo():
         outputs=[run_btn, logs, uid_label],
     )
 
-    def enable_run_btn():
+    def enable_run_btn(run_inference, inference_token, model_id, dataset_id, dataset_config, dataset_split):
+        if run_inference and inference_token == "":
+            return gr.update(interactive=False)
+        if model_id == "" or dataset_id == "" or dataset_config == "" or dataset_split == "":
+            return gr.update(interactive=False)
         return gr.update(interactive=True)
 
     gr.on(
@@ -246,13 +252,27 @@ def get_demo():
             scanners.input,
         ],
         fn=enable_run_btn,
-        inputs=None,
+        inputs=[
+            run_inference, 
+            inference_token, 
+            model_id_input, 
+            dataset_id_input, 
+            dataset_config_input, 
+            dataset_split_input
+        ],
         outputs=[run_btn],
     )
 
     gr.on(
         triggers=[label.input for label in column_mappings],
         fn=enable_run_btn,
-        inputs=None,  # FIXME
+        inputs=[
+            run_inference, 
+            inference_token, 
+            model_id_input, 
+            dataset_id_input, 
+            dataset_config_input, 
+            dataset_split_input
+        ],  # FIXME
         outputs=[run_btn],
     )

text_classification_ui_helpers.py

@@ -215,7 +215,7 @@ def precheck_model_ds_enable_example_btn(
 
 
 def align_columns_and_show_prediction(
-    model_id, dataset_id, dataset_config, dataset_split, uid
+    model_id, dataset_id, dataset_config, dataset_split, uid, run_inference, inference_token
 ):
     ppl = check_model(model_id)
     if ppl is None or not isinstance(ppl, TextClassificationPipeline):
@@ -276,7 +276,7 @@ def align_columns_and_show_prediction(
             gr.update(value=MAPPING_STYLED_ERROR_WARNING, visible=True),
             gr.update(visible=False),
             gr.update(visible=True, open=True),
-            gr.update(interactive=True),
+            gr.update(interactive=(run_inference and inference_token != "")),
             "",
             *column_mappings,
         )
@@ -288,7 +288,7 @@ def align_columns_and_show_prediction(
         gr.update(value=get_styled_input(prediction_input), visible=True),
         gr.update(value=prediction_output, visible=True),
         gr.update(visible=True, open=False),
-        gr.update(interactive=True),
+        gr.update(interactive=(run_inference and inference_token != "")),
         "",
         *column_mappings,
     )