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 ) |