patrickvonplaten
commited on
Commit
•
090ea4d
1
Parent(s):
da69cd8
add update model card script
Browse files- README.md +10 -2
- update_model_card_mass_pr.py +12 -0
README.md
CHANGED
@@ -1,4 +1,12 @@
|
|
1 |
-
#
|
|
|
|
|
|
|
2 |
|
3 |
Read https://x-lance.sjtu.edu.cn/papers/zhc00-chen-icassp17.pdf
|
4 |
-
|
|
|
|
|
|
|
|
|
|
1 |
+
# This repo contains some useful scripts
|
2 |
+
|
3 |
+
|
4 |
+
## Confidence Scoring
|
5 |
|
6 |
Read https://x-lance.sjtu.edu.cn/papers/zhc00-chen-icassp17.pdf
|
7 |
+
|
8 |
+
Run `create_confidence_scores.py`
|
9 |
+
|
10 |
+
## Mass PR to update README
|
11 |
+
|
12 |
+
Run `update_model_card_mass_pr.py` . Make sure you use the following branch: https://github.com/huggingface/huggingface_hub/pull/884
|
update_model_card_mass_pr.py
ADDED
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
#!/usr/bin/env python3
|
2 |
+
from huggingface_hub import hf_api
|
3 |
+
from huggingface_hub.repocard import metadata_update
|
4 |
+
|
5 |
+
hf_api = hf_api.HfApi()
|
6 |
+
|
7 |
+
all_marian_models = hf_api.list_models(filter="marian")
|
8 |
+
all_new_marian_models = [x.modelId for x in all_marian_models if x.modelId.startswith("Helsinki-NLP/opus-mt-tc")]
|
9 |
+
|
10 |
+
for repo_id in all_new_marian_models:
|
11 |
+
print(f"Opening PR for {repo_id}")
|
12 |
+
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")
|