Update README.md

README.md

@@ -30,11 +30,17 @@ Determine whether online sales products are adult content. Input: image content,
 
 # use transformers pipeline
 ```python
-from transformers import pipeline
-pipe = pipeline("image-classification", model="jiechau/adult-content-classifier")
+from transformers import pipeline, AutoConfig
+pipe = pipeline("image-classification", model="jiechau/adult-content-identify-image")
+config = AutoConfig.from_pretrained("jiechau/adult-content-identify-image")
+label2id = config.label2id
+id2label = config.id2label
+
 q = 'https://xxx.xxx.xxx/images/xxx/xxx.webp'
 q = 'https://xxx.xxx.xxx/images/xxx/xxx.jpg'
 result = pipe(q)
 print(result)
+print(label2id[result[0]['label']])
 # [{'label': 'adult_成人商品', 'score': 0.7516837120056152}, {'label': 'regular_一般商品', 'score': 0.2475457787513733}, {'label': 'unknown', 'score': 0.0007705678581260145}]
+# 1
 ```