Views
No views yet
1from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
2
3tokenizer = AutoTokenizer.from_pretrained("lamaachi/distilbert-moroccan-pii-classifier")
4model = AutoModelForTokenClassification.from_pretrained("your-username/distilbert-moroccan-pii-classifier")
5
6nlp = pipeline("ner", model=model, tokenizer=tokenizer)
7
8text = "Smiya dyali Youssef, num 0612345678, email: test@gmail.com"
9result = nlp(text)
10
11print(result)