Add feature mapping and label mapping

app.py

@@ -152,7 +152,15 @@ def try_validate(model_id, dataset_id, dataset_config, dataset_split, column_map
     )
 
 
-def try_submit(m_id, d_id, config, split, local):
+def try_submit(m_id, d_id, config, split, column_mappings, local):
+    label_mapping = {}
+    try:
+        column_mapping = json.loads(column_mappings)
+        if "label" in column_mapping:
+            label_mapping = column_mapping.pop("label", {})
+    except Exception:
+        column_mapping = {}
+
     if local:
         command = [
             "python",
@@ -166,6 +174,8 @@ def try_submit(m_id, d_id, config, split, local):
             "--discussion_repo", os.environ.get(HF_REPO_ID) or os.environ.get(HF_SPACE_ID),
             "--output_format", "markdown",
             "--output_portal", "huggingface",
+            "--feature_mapping", json.dumps(column_mapping),
+            "--label_mapping", json.dumps(label_mapping),
         ]
 
         eval_str = f"[{m_id}]<{d_id}({config}, {split} set)>"
@@ -273,6 +283,7 @@ with gr.Blocks(theme=theme) as iface:
                 dataset_id_input,
                 dataset_config_input,
                 dataset_split_input,
+                column_mapping_input,
                 run_local,
             ],
         )