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

CowCowC/Adu_mod_id_img

adult-content-identify-image

use transformers pipeline

CowCowC/
Adu_mod_id_img