datasetsANDmodels
/

occupation-extraction

text2text-generation

Generated from Trainer

text-generation-inference

Model card Files Files and versions

Metrics Training metrics Community

datasetsANDmodels commited on Aug 8

Commit

bb98627

·

1 Parent(s): 86a3875

Update usage.py

Files changed (1) hide show

usage.py +1 -1

usage.py CHANGED Viewed

@@ -1,6 +1,6 @@
 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=="":

 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=="":