jiechau's picture
Update README.md
f5df8e8 verified
|
raw
history blame
1.44 kB
metadata
tags:
  - generated_from_keras_callback
model-index:
  - name: adult-content-classifier-text
    results: []
pipeline_tag: text-classification
widget:
  - text: 情趣睡衣 性感睡衣 角色扮演 惹火 緊身連身 漆皮膠衣  L
    example_title: adult_成人商品標題
  - text: 男平織運動外套-立領外套 慢跑 路跑 藍黑螢光黃 L
    example_title: regular_一般商品標題

adult-content-identify-text

Determine whether online sales products are adult content. Input: product title or description, Output results: 0 Unknown, 1 Adult Content, 2 General Merchandise.

判斷網路銷售商品是否屬於成人內容。輸入: 商品名稱文字,輸出結果: 0 未知, 1 成人內容, 2 一般商品。

use transformers pipeline

from transformers import pipeline
pipe = pipeline("text-classification", model="jiechau/adult-content-classifier")
#q = '男平織運動外套-立領外套 慢跑 路跑 藍黑螢光黃 L'
q = '情趣睡衣 性感睡衣 角色扮演 惹火 緊身連身 漆皮膠衣 黑 L'
result = pipe(q)
print(result)
# [{'label': 'adult_成人商品', 'score': 0.9994813799858093}]

Framework versions

  • Transformers 4.37.1
  • TensorFlow 2.15.0
  • Datasets 2.17.0
  • Tokenizers 0.15.1