File size: 287 Bytes
75722f1
 
b653e77
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
import websockets
from transformers import pipeline
extractor = pipeline("text2text-generation", model="occ_extract")
intent  ="I need a lawn mower for my garden"
label=extractor(intent)[0]["generated_text"]
if label=="":
               label="No occupation detected"
print (label )