import websockets from transformers import pipeline extractor = pipeline("text2text-generation", model="datasetsANDmodels/occupation-extraction") intent ="I need a lawn mower for my garden" label=extractor(intent)[0]["generated_text"] if label=="": label="No occupation detected" print (label )