datasetsANDmodels commited on
Commit
c4744a2
·
1 Parent(s): 36af45f

Create usage.py

Browse files
Files changed (1) hide show
  1. usage.py +7 -0
usage.py ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+ extractor = pipeline("text2text-generation", model="task-extraction")
3
+ intent ="I am learning French."
4
+ label=extractor(intent)[0]["generated_text"]
5
+ if label=="":
6
+ label="No task detected"
7
+ print (label )