Extract label mappings from table

app.py

@@ -160,12 +160,10 @@ def try_validate(m_id, ppl, dataset_id, dataset_config, dataset_split, column_ma
 
 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 = {}
+    for i, label in column_mappings["Model Prediction Labels"].iteritems():
+        label_mapping.update({str(i): label})
+
+    # TODO: Set column mapping for some dataset such as `amazon_polarity`
 
     if local:
         command = [
@@ -180,7 +178,7 @@ def try_submit(m_id, d_id, config, split, column_mappings, 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),
+            # TODO: "--feature_mapping", json.dumps(column_mapping),
             "--label_mapping", json.dumps(label_mapping),
         ]