🐛 fix small model card bugs (huggingface#13310)

nateraw · web-flow · commit ce6add8ecc88 · 2021-08-30T08:45:57.000-06:00
* 🐛 fix small model card bugs

* 💄 style
diff --git a/src/transformers/modelcard.py b/src/transformers/modelcard.py
@@ -426,12 +426,15 @@ def create_model_index(self, metric_mapping):
                     result["dataset"]["args"] = dataset_arg_mapping[ds_tag]
 
             if len(metric_mapping) > 0:
+                result["metrics"] = []
                 for metric_tag, metric_name in metric_mapping.items():
-                    result["metric"] = {
-                        "name": metric_name,
-                        "type": metric_tag,
-                        "value": self.eval_results[metric_name],
-                    }
+                    result["metrics"].append(
+                        {
+                            "name": metric_name,
+                            "type": metric_tag,
+                            "value": self.eval_results[metric_name],
+                        }
+                    )
 
             model_index["results"].append(result)
 
@@ -446,7 +449,7 @@ def create_metadata(self):
         metadata = _insert_values_as_list(metadata, "tags", self.tags)
         metadata = _insert_values_as_list(metadata, "datasets", self.dataset_tags)
         metadata = _insert_values_as_list(metadata, "metrics", list(metric_mapping.keys()))
-        metadata["model_index"] = self.create_model_index(metric_mapping)
+        metadata["model-index"] = self.create_model_index(metric_mapping)
 
         return metadata
 
