kryman27 commited on
Commit
0f572b1
·
verified ·
1 Parent(s): 5a1bc01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import re
4
  from transformers import pipeline
5
 
6
  # Model do rozpoznawania nazw organizacji i wartości numerycznych
7
- extractor = pipeline("ner", model="dbmdz/bert-large-cased-finetuned-conll03-english", aggregation_strategy="simple")
8
 
9
  # Reguły do wykrywania NIP, kwot, dat
10
  nip_pattern = re.compile(r'\bPL\s?\d{10}\b|\b\d{10}\b') # Polski NIP (z "PL" lub bez)
 
4
  from transformers import pipeline
5
 
6
  # Model do rozpoznawania nazw organizacji i wartości numerycznych
7
+ extractor = pipeline("ner", model="kryman27/layoutlmv3-finetuned", aggregation_strategy="simple")
8
 
9
  # Reguły do wykrywania NIP, kwot, dat
10
  nip_pattern = re.compile(r'\bPL\s?\d{10}\b|\b\d{10}\b') # Polski NIP (z "PL" lub bez)