Views
No views yet
1from transformers import pipeline, AutoConfig
2pipe = pipeline("text-classification", model="jiechau/adult-content-identify-text")
3config = AutoConfig.from_pretrained("jiechau/adult-content-identify-text")
4label2id = config.label2id
5id2label = config.id2label
6
7#q = 'Men's Plain Weave Sports Jacket - Stand Collar Jacket for Jogging and Running, Blue-Black-Fluorescent Yellow, Size L'
8#q = 'Passionate Embrace: Cross-Laced Sexy Silky Lingerie Sleepwear, White'
9#q = '男平織運動外套-立領外套 慢跑 路跑 藍黑螢光黃 L'
10q = '情趣睡衣 性感睡衣 角色扮演 惹火 緊身連身 漆皮膠衣 黑 L'
11result = pipe(q)
12print(result)
13print(label2id[result[0]['label']])
14# [{'label': 'adult_成人商品', 'score': 0.9994813799858093}]
15# 1