Update text_classification_ui_helpers.py

text_classification_ui_helpers.py

@@ -298,7 +298,8 @@ def construct_label_and_feature_mapping(all_mappings):
     for i, label in zip(
         range(len(all_mappings["labels"].keys())), all_mappings["labels"].keys()
     ):
-        label_mapping.update({str(i): label})
+        # FIXME: What's the order during the save
+        label_mapping.update({str(i): all_mappings["labels"][label]})
 
     if "features" not in all_mappings.keys():
         gr.Warning(CONFIRM_MAPPING_DETAILS_FAIL_RAW)