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