purpose-extraction / usage.py
datasetsANDmodels's picture
Update usage.py
bc210a2
from transformers import pipeline
extractor = pipeline("text2text-generation", model="purpose-extraction")
intent = " I am calling to invite you the meeting."
label=extractor(intent)[0]["generated_text"]
if label=="":
label="No purpose detected"
print (label )