jiechau's picture
Update README.md
e757b3c verified
|
raw
history blame
1.07 kB
metadata
tags:
  - generated_from_keras_callback
model-index:
  - name: adult-content-classifier
    results: []
pipeline_tag: text-classification
widget:
  - text: 情趣睡衣 性感睡衣 角色扮演 惹火 緊身連身 漆皮膠衣  L
    example_title: 商品標題1
  - text: 男平織運動外套-立領外套 慢跑 路跑 藍黑螢光黃 L
    example_title: 商品標題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': '成人商品', 'score': 0.9994813799858093}]

Framework versions

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