datasetsANDmodels commited on
Commit
18629fa
·
verified ·
1 Parent(s): 707824c

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= datasetsANDmodels/time-extraction")
3
+ intent = "How about scheduling a project update meeting on the first Monday of November at 1:30 PM?"
4
+ label=extractor(intent)[0]["generated_text"]
5
+ if label=="":
6
+ label="No time detected"
7
+ print (label )