questionClassifier / README.md
paulbauriegel's picture
Create README.md
ced43ed
|
raw
history blame
241 Bytes
from transformers import pipeline

pipe = pipeline(model="hamzaMM/questionClassifier", task="text-classification")
def is_question(input: str):
    return pipe(input)[0]['label'] == 'LABEL_1'

is_question("How is the weather")