How to use this classifier:
from transformers import pipeline
pipe = pipeline("text-classification", model="Peterard/distilbert_feature_classifier")
pipe("Please add a like button!")
# [{'label': 'feature_request', 'score': 0.8930749893188477}]
pipe("The app crashed when I opened it this morning. Can you fix this please?")
#[{'label': 'no_feature_request', 'score': 0.9971746206283569}]
N.B. The label will change depending on which is the likelier class
- Downloads last month
- 14