add styled error

text_classification_ui_helpers.py

@@ -23,7 +23,7 @@ from wordings import (
     CONFIRM_MAPPING_DETAILS_FAIL_RAW,
     MAPPING_STYLED_ERROR_WARNING,
     NOT_TEXT_CLASSIFICATION_MODEL_RAW,
-    UNMATCHED_MODEL_DATASET,
+    UNMATCHED_MODEL_DATASET_STYLED_ERROR,
     CHECK_LOG_SECTION_RAW,
     get_styled_input,
 )
@@ -255,9 +255,9 @@ def align_columns_and_show_prediction(
         )
     
     if len(ds_labels) != len(model_labels):
-        gr.Warning(UNMATCHED_MODEL_DATASET)
+        # gr.Warning(UNMATCHED_MODEL_DATASET)
         return (
-            gr.update(visible=False),
+            gr.update(value=UNMATCHED_MODEL_DATASET_STYLED_ERROR, visible=True),
             gr.update(visible=False),
             gr.update(visible=False, open=False),
             gr.update(interactive=False),