occupation-extractor / occ-use.py
datasetsANDmodels's picture
Upload occ-use.py
d03d62b verified
import websockets
from transformers import pipeline
extractor = pipeline("text2text-generation", model="occ")
intent ="I need a lawn mower for my garden"
label=extractor(intent)[0]["generated_text"]
if label=="":
label="No occupation detected"
print (label )