Update README.md

README.md

@@ -35,21 +35,27 @@ probably proofread and complete it, then remove this comment. -->
 
 # adult-content-identify-text
 
-Support languages: English, Chinese, Japanese, Korean
+Support languages: English, Chinese (both CN/TW), Japanese, Korean
 
-Determine whether online sales products are adult content. Input: product title or description, Output results: 0 Unknown, 1 Adult Content, 2 General Merchandise.
+Determine whether online sales products are adult content. Input: text of product title or description, Output results 0: Unknown, 1: Adult Content, 2: General Merchandise.
 
 判斷網路銷售商品是否屬於成人內容。輸入: 商品名稱文字，輸出結果: 0 未知, 1 成人內容, 2 一般商品。
 
 # use transformers pipeline
 ```python
-from transformers import pipeline
-pipe = pipeline("text-classification", model="jiechau/adult-content-classifier")
+from transformers import pipeline, AutoConfig
+pipe = pipeline("text-classification", model="jiechau/adult-content-identify-text")
+config = AutoConfig.from_pretrained("jiechau/adult-content-identify-text")
+label2id = config.label2id
+id2label = config.id2label
+
 #q = '男平織運動外套-立領外套 慢跑 路跑 藍黑螢光黃 L'
 q = '情趣睡衣 性感睡衣 角色扮演 惹火 緊身連身 漆皮膠衣 黑 L'
 result = pipe(q)
 print(result)
+print(label2id[result[0]['label']])
 # [{'label': 'adult_成人商品', 'score': 0.9994813799858093}]
+# 1
 ```
 
 ### Framework versions