from transformers import pipeline extractor = pipeline("text2text-generation", model="task-extraction") label=extractor(intent)[0]["generated_text"] if label=="": label="No task detected" print (intent.strip(),"->",label )