add onnx tag

app.py

@@ -40,7 +40,7 @@ def onnx_export(
 
         api = HfApi(token=oauth_token.token)
 
-        error, commit_info = convert(api=api, model_id=model_id, task=task, opset=opset)
+        error, commit_info = convert(api=api, model_id=model_id, task=task, opset=opset, token=oauth_token.token)
         if error != "0":
             return error
 

onnx_export.py

@@ -6,6 +6,9 @@ from typing import List, Optional, Tuple
 from huggingface_hub import (
     CommitOperationAdd,
     HfApi,
+    ModelCard,
+    Discussion,
+    CommitInfo,
 )
 from huggingface_hub.file_download import repo_folder_name
 from optimum.exporters.onnx import main_export
@@ -27,12 +30,13 @@ def previous_pr(api: "HfApi", model_id: str, pr_title: str) -> Optional["Discuss
             return discussion
 
 
-def export_and_git_add(model_id: str, task: str, folder: str, opset: int) -> List:
+def export_and_git_add(model_id: str, task: str, folder: str, opset: int, token: str) -> List:
     main_export(
         model_name_or_path=model_id,
         output=folder,
         task=task,
         opset=opset,
+        token=token,
     )
 
     n_files = len(
@@ -59,6 +63,16 @@ def export_and_git_add(model_id: str, task: str, folder: str, opset: int) -> Lis
             for file_name in os.listdir(folder)
         ]
 
+    try:
+        card = ModelCard.load(model_id, token=token)
+        if card.data.tags is None:
+            card.data.tags = []
+        card.data.tags.append("onnx")
+        card.save(os.path.join(folder, "README.md"))
+        operations.append(CommitOperationAdd(path_in_repo="README.md", path_or_fileobj=os.path.join(folder, "README.md")))
+    except:
+        pass
+
     return operations
 
 
@@ -68,6 +82,7 @@ def convert(
     task: str,
     force: bool = False,
     opset: int = None,
+    token: str = None,
 ) -> Tuple[int, "CommitInfo"]:
     pr_title = "Adding ONNX file of this model"
     info = api.model_info(model_id)
@@ -92,7 +107,7 @@ def convert(
                     f"Model {model_id} already has an open PR check out [{url}]({url})"
                 )
             else:
-                operations = export_and_git_add(model_id, task, folder, opset)
+                operations = export_and_git_add(model_id, task, folder, opset, token)
 
                 commit_description = f"""
 Beep boop I am the [ONNX export bot 🤖🏎️]({SPACES_URL}). On behalf of [{requesting_user}](https://huggingface.co/{requesting_user}), I would like to add to this repository the model converted to ONNX.