File size: 237 Bytes
d78ba77 |
1 2 3 4 5 6 7 |
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 )
|