fix check validity ui

app_text_classification.py

@@ -8,13 +8,13 @@ from text_classification_ui_helpers import (
     align_columns_and_show_prediction,
     check_dataset,
     precheck_model_ds_enable_example_btn,
+    show_hf_token_info,
     try_submit,
     write_column_mapping_to_config,
 )
 
 from text_classification import (
   get_example_prediction, 
-  check_hf_token_validity, 
   HuggingFaceInferenceAPIResponse
 )
 from wordings import (
@@ -104,7 +104,7 @@ def get_demo():
         inference_token_info = gr.HTML(value=HF_TOKEN_INVALID_STYLED, visible=False)
 
         inference_token.change(
-            lambda token: gr.update(visible=lambda: check_hf_token_validity(token)),
+            fn=show_hf_token_info,
             inputs=[inference_token],
             outputs=[inference_token_info],
         )
@@ -244,11 +244,7 @@ def get_demo():
             return gr.update(interactive=False)
         if not column_mapping_accordion.visible:
             return gr.update(interactive=False)
-        _, prediction_response = get_example_prediction(
-            model_id, dataset_id, dataset_config, dataset_split, inference_token
-        )
-        if not isinstance(prediction_response, HuggingFaceInferenceAPIResponse):
-            gr.warning("Your HF token is invalid. Please check your token.")
+        if inference_token_info.visible:
             return gr.update(interactive=False)
         return gr.update(interactive=True)
 

text_classification_ui_helpers.py

@@ -16,6 +16,7 @@ from text_classification import (
     preload_hf_inference_api,
     get_example_prediction,
     get_labels_and_features_from_dataset,
+    check_hf_token_validity,
     HuggingFaceInferenceAPIResponse,
 )
 from wordings import (
@@ -331,6 +332,12 @@ def construct_label_and_feature_mapping(all_mappings, ds_labels, ds_features):
     feature_mapping = all_mappings["features"]
     return label_mapping, feature_mapping
 
+def show_hf_token_info(token):
+    valid = check_hf_token_validity(token)
+    if not valid:
+        return gr.update(visible=True)
+    return gr.update(visible=False)
+
 def try_submit(m_id, d_id, config, split, inference, inference_token, uid):
     all_mappings = read_column_mapping(uid)
     check_column_mapping_keys_validity(all_mappings)