| #!/usr/bin/env python3 | |
| from huggingface_hub import hf_api | |
| from huggingface_hub.repocard import metadata_update | |
| hf_api = hf_api.HfApi() | |
| all_marian_models = hf_api.list_models(filter="marian") | |
| all_new_marian_models = [x.modelId for x in all_marian_models if x.modelId.startswith("Helsinki-NLP/opus-mt-tc")] | |
| for repo_id in all_new_marian_models: | |
| print(f"Opening PR for {repo_id}") | |
| metadata_update(repo_id, {"tags": ["translation", "opus-mt-tc"]}, create_pr=True, pr_commit_summary="Add `opus-mt-tc` tag", pr_commit_description="Add `opus-mt-tc` tag so that this new marian models can be found more easily") | |