Show check config warning

text_classification_ui_helpers.py

@@ -12,7 +12,7 @@ from io_utils import (get_yaml_path, read_column_mapping, save_job_to_pipe,
                       write_column_mapping, write_log_to_user_file)
 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
+from wordings import CONFIRM_MAPPING_DETAILS_FAIL_RAW, MAPPING_STYLED_ERROR_WARNING, CHECK_CONFIG_OR_SPLIT_RAW
 
 MAX_LABELS = 20
 MAX_FEATURES = 20
@@ -131,7 +131,7 @@ def check_model_and_show_prediction(
 
     # when dataset does not have labels or features
     if not isinstance(ds_labels, list) or not isinstance(ds_features, list):
-        # gr.Warning(CONFIRM_MAPPING_DETAILS_FAIL_RAW)
+        gr.Warning(CHECK_CONFIG_OR_SPLIT_RAW)
         return (
             gr.update(visible=False),
             gr.update(visible=False),

wordings.py

@@ -21,6 +21,10 @@ CONFIRM_MAPPING_DETAILS_FAIL_RAW = """
                             Sorry, we cannot align the input/output of your dataset with the model. Pleaser double check your model and dataset.
                             """
 
+CHECK_CONFIG_OR_SPLIT_RAW = """
+                            Please check your dataset config or split.
+                            """
+
 PREDICTION_SAMPLE_MD = """
                         <h1 style="text-align: center;">
                         Model Prediction Sample