datasetsANDmodels's picture
Update usage.py
0f7d53a verified
raw
history blame contribute delete
293 Bytes
from transformers import pipeline
extractor = pipeline("text2text-generation", model="datasetsANDmodels/location-extraction")
intent = " I will visit you at botanic garden."
label=extractor(intent)[0]["generated_text"]
if label=="":
label="No location detected"
print (label )