File size: 250 Bytes
c4744a2
 
 
 
 
 
 
1
2
3
4
5
6
7
from transformers import pipeline
extractor = pipeline("text2text-generation", model="task-extraction")
intent  ="I am learning French."
label=extractor(intent)[0]["generated_text"]
if label=="":
               label="No task detected"
print (label )