Fix label mapping

text_classification_ui_helpers.py

@@ -183,7 +183,9 @@ def try_submit(m_id, d_id, config, split, local, uid):
     if "labels" not in all_mappings.keys():
         gr.Warning(CONFIRM_MAPPING_DETAILS_FAIL_RAW)
         return (gr.update(interactive=True), gr.update(visible=False))
-    label_mapping = all_mappings["labels"]
+    label_mapping = {}
+    for i, label in zip(range(len(all_mappings["labels"].keys())), all_mappings["labels"].keys()):
+        label_mapping.update({str(i): label})
 
     if "features" not in all_mappings.keys():
         gr.Warning(CONFIRM_MAPPING_DETAILS_FAIL_RAW)