request-extraction / usage.py
datasetsANDmodels's picture
Update usage.py
8728d75
raw
history blame contribute delete
361 Bytes
from transformers import pipeline
extractor = pipeline("text2text-generation", model="request-extraction")
intent ="I really need my broken window to be replaced, can John do it by today"
intent = " Can you come and have a look at our boiler"
label=extractor(intent)[0]["generated_text"]
if label=="":
label="No request detected"
print (label )