File size: 315 Bytes
75722f1 bb98627 b653e77 |
1 2 3 4 5 6 7 8 9 |
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 )
|