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 )