Commit e6884fa
committed
improvement(pii): assert real GLiNER inference in the image build
The bake step previously only proved GLiNER.from_pretrained loads —
a transformers/gliner bump that silently stops returning detections
would still build and pass /health. The bake step now runs
predict_entities on a sample sentence (labels from engines.py's
GLINER_ENTITY_MAPPING) and fails the build unless person + location
are detected.
Verified against the exact bumped pin set (transformers 5.5.0,
huggingface_hub 1.23.0, gliner 0.2.27, torch 2.11.0): person 0.9999,
location 0.9951, date 0.9987.1 parent b3eb3b0 commit e6884fa
1 file changed
Lines changed: 8 additions & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
87 | 87 | | |
88 | 88 | | |
89 | 89 | | |
| 90 | + | |
| 91 | + | |
| 92 | + | |
90 | 93 | | |
91 | 94 | | |
92 | | - | |
| 95 | + | |
| 96 | + | |
| 97 | + | |
| 98 | + | |
| 99 | + | |
93 | 100 | | |
94 | 101 | | |
95 | 102 | | |
| |||
0 commit comments