Hugging Face's logo

Spaces:
Ammar-alhaj-ali
/
LayoutLMv3-Invoice
Runtime error

App Files Community
Ammar-alhaj-ali commited on
Commit
8587e07
•
1 Parent(s): 4dd4b40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,7 +32,7 @@ labels = ['O', 'B-ABN', 'B-BILLER', 'B-BILLER_ADDRESS', 'B-BILLER_POST_CODE', 'B
32
  id2label = {v: k for v, k in enumerate(labels)}
33
  label2color = {
34
  "B-ABN": 'blue',
35
- "B-BILLER": 'blue',
36
  "B-BILLER_ADDRESS": 'green',
37
  "B-BILLER_POST_CODE": 'orange',
38
  "B-DUE_DATE": "blue",
@@ -40,7 +40,7 @@ label2color = {
40
  "B-INVOICE_DATE": 'violet',
41
  "B-INVOICE_NUMBER": 'orange',
42
  "B-SUBTOTAL": 'green',
43
- "B-TOTAL": 'blue',
44
  "I-BILLER_ADDRESS": 'blue',
45
  "O": 'orange'
46
  }
 
32
  id2label = {v: k for v, k in enumerate(labels)}
33
  label2color = {
34
  "B-ABN": 'blue',
35
+ "B-BILLER": 'red',
36
  "B-BILLER_ADDRESS": 'green',
37
  "B-BILLER_POST_CODE": 'orange',
38
  "B-DUE_DATE": "blue",
 
40
  "B-INVOICE_DATE": 'violet',
41
  "B-INVOICE_NUMBER": 'orange',
42
  "B-SUBTOTAL": 'green',
43
+ "B-TOTAL": 'red',
44
  "I-BILLER_ADDRESS": 'blue',
45
  "O": 'orange'
46
  }