request-extraction / usage.py
datasetsANDmodels's picture
Create usage.py
c4744a2
raw
history blame
250 Bytes
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 )