time-extraction / usage.py
datasetsANDmodels's picture
Create usage.py
18629fa verified
from transformers import pipeline
extractor = pipeline("text2text-generation", "model= datasetsANDmodels/time-extraction")
intent = "How about scheduling a project update meeting on the first Monday of November at 1:30 PM?"
label=extractor(intent)[0]["generated_text"]
if label=="":
label="No time detected"
print (label )