| """Check the install: find and redact PII in four sample texts (Hindi, romanised Hindi, Tamil, Bengali; all values | |
| are invented). | |
| python example.py | |
| """ | |
| from pii import PIIExtractor | |
| SAMPLES = [ | |
| "नमस्ते, मेरा नाम अनीता कुलकर्णी है। मेरा मोबाइल नंबर 98204 51763 है और ईमेल anita.k@example.in है।", | |
| "Sir, mera PAN BXQPK4821M hai aur account number 50100234567812, IFSC HDFC0004521. Kripya update kar dijiye.", | |
| "என் பெயர் முருகன் செல்வம். என் ஆதார் எண் 4521 8890 2316, முகவரி 14, காந்தி தெரு, மதுரை 625001.", | |
| "আমার UPI আইডি rakesh.das@okaxis, গাড়ির নম্বর WB 02 AK 4417। অর্ডার নম্বর ORD-55231 এখনও আসেনি।", | |
| ] | |
| pii = PIIExtractor() | |
| print(f"model loaded on {pii.device}\n") | |
| for text in SAMPLES: | |
| spans, ms = pii.timed_extract(text) | |
| print(text) | |
| for s in spans: | |
| print(f" {s['label']:15} {s['text']!r} [{s['start']}:{s['end']}]") | |
| print(f" redacted: {pii.redact(text)}") | |
| print(f" ({ms:.0f} ms)\n") | |