datasetsANDmodels commited on
Commit
d78ba77
·
verified ·
1 Parent(s): fe19188

Create usage.py

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