from gliner import GLiNER
model = GLiNER.from_pretrained("urchade/gliner_mediumv2.1")
model.eval()
def identify_pii_and_mask(text,labels):
entities = model.predict_entities(text, labels, threshold=0.4)
return entities
how to save and build docker image for this model
could you please any one help this
as i know the first step is regiseter the model then serve the model and bentoml build as it is a transformer model i am not geeting how to do this
could you please anyone help with this