Metadata update (huggingface#23259)

* Metadata update * Make fixup
fenglui · May 10, 2023 · 4f05bbf · 4f05bbf
1 parent 996f127
commit 4f05bbf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/transformers/tools/base.py b/src/transformers/tools/base.py
@@ -23,7 +23,7 @@
 import tempfile
 from typing import Any, Dict, List, Optional, Union
 
-from huggingface_hub import CommitOperationAdd, HfFolder, create_commit, create_repo, hf_hub_download
+from huggingface_hub import CommitOperationAdd, HfFolder, create_commit, create_repo, hf_hub_download, metadata_update
 from huggingface_hub.utils import RepositoryNotFoundError, get_session
 
 from ..dynamic_module_utils import custom_object_save, get_class_from_dynamic_module, get_imports
@@ -285,6 +285,7 @@ def push_to_hub(
         repo_url = create_repo(
             repo_id=repo_id, token=token, private=private, exist_ok=True, repo_type="space", space_sdk="gradio"
         )
+        metadata_update(repo_id, {"tags": ["tool"]}, repo_type="space")
         repo_id = repo_url.repo_id
 
         with tempfile.TemporaryDirectory() as work_dir: