fix test token bug and add icons to wording

text_classification.py

@@ -389,7 +389,8 @@ def check_hf_token_validity(hf_token):
         return False
     # use inference api to check the token
     payload = {"inputs": "This is a test", "options": {"use_cache": True}}
-    response = hf_inference_api("cardiffnlp/twitter-roberta-base-sentiment-latest", hf_token, payload)
+    # use a tiny model to test the token
+    response = hf_inference_api("cross-encoder/ms-marco-TinyBERT-L-2-v2", hf_token, payload)
     if "error" in response:
         return False
     return True

text_classification_ui_helpers.py

@@ -329,11 +329,8 @@ def enable_run_btn(uid, run_inference, inference_token, model_id, dataset_id, da
         logger.warn("Column mapping is not valid")
         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):
-        logger.warn("Failed to get example prediction from inference API with this token")
+    if not check_hf_token_validity(inference_token):
+        logger.warn("HF token is not valid")
         return gr.update(interactive=False)
     return gr.update(interactive=True)
 

wordings.py

@@ -38,13 +38,13 @@ PREDICTION_SAMPLE_MD = """
 
 MAPPING_STYLED_ERROR_WARNING = """
                         <h3 style="text-align: center;color: orange; background-color: #fff0f3; border-radius: 8px; padding: 10px; ">
-                        Sorry, we cannot auto-align the labels/features of your dataset and model. Please double check.
+                        ⚠️ We cannot auto-align the labels/features of your dataset and model. Please double check the info below before your submission.
                         </h3>
                         """
 
 UNMATCHED_MODEL_DATASET_STYLED_ERROR = """
                         <h3 style="text-align: center;color: #fa5f5f; background-color: #fbe2e2; border-radius: 8px; padding: 10px; ">
-                        Your model and dataset have different numbers of labels. Please double check your model and dataset.
+                        ❌ Your model and dataset have different numbers of labels. Please double check your model and dataset.
                         </h3>
                         """