SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Use repo_type instead of deprecated datasets repo IDs (#19202)

Browse Source 
* Use repo_type instead of deprecated datasets repo IDs

* Add missing one in doc
This commit is contained in:
Sylvain Gugger
2022-09-26 09:50:48 -04:00
committed by GitHub
parent 216b2f9e80
commit c20b2c7e18
32 changed files with 74 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/source/en/tasks/semantic_segmentation.mdx
View File

@@ -67,9 +67,9 @@ You'll also want to create a dictionary that maps a label id to a label class wh
>>> import json
>>> from huggingface_hub import cached_download, hf_hub_url
>>> repo_id = "datasets/huggingface/label-files"
>>> repo_id = "huggingface/label-files"
>>> filename = "ade20k-id2label.json"
>>> id2label = json.load(open(cached_download(hf_hub_url(repo_id, filename)), "r"))
>>> id2label = json.load(open(cached_download(hf_hub_url(repo_id, filename, repo_type="dataset")), "r"))
>>> id2label = {int(k): v for k, v in id2label.items()}
>>> label2id = {v: k for k, v in id2label.items()}
>>> num_labels = len(id2label)