add persistent error code when number of labels not matching

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,
 )
@@ -257,9 +257,8 @@ def align_columns_and_show_prediction(
         )
     
     if len(ds_labels) != len(model_labels):
-        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),

wordings.py

@@ -17,10 +17,6 @@ CONFIRM_MAPPING_DETAILS_FAIL_MD = """
                             Sorry, we cannot align the input/output of your dataset with the model. <b>Pleaser double check your model and dataset.</b>
                             """
 
-UNMATCHED_MODEL_DATASET = """
-                            Model prediction labels do not align with the labels present in the dataset. Please double check your model and dataset.
-                            """
-
 CONFIRM_MAPPING_DETAILS_FAIL_RAW = """
                             Sorry, we cannot auto-align the input/output of your dataset with the model.
                             """
@@ -41,11 +37,17 @@ PREDICTION_SAMPLE_MD = """
                         """
 
 MAPPING_STYLED_ERROR_WARNING = """
-                        <h3 style="text-align: center;color: coral; background-color: #fff0f3; border-radius: 8px; padding: 10px; ">
+                        <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.
                         </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.
+                        </h3>
+                        """
+
 NOT_TEXT_CLASSIFICATION_MODEL_RAW = """
                         Your model does not fall under the category of text classification. This page is specifically designated for the evaluation of text classification models.  
                       """