omarelsayeed commited on
Commit
20a0ec6
·
verified ·
1 Parent(s): e996bd4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ import torch
11
  from transformers import LayoutLMv3ForTokenClassification
12
 
13
  # Load the LayoutLMv3 model
14
- layout_model = LayoutLMv3ForTokenClassification.from_pretrained("omarelsayeed/LayoutReader80Small")
15
 
16
  MAX_LEN = 100
17
  CLS_TOKEN_ID = 0
 
11
  from transformers import LayoutLMv3ForTokenClassification
12
 
13
  # Load the LayoutLMv3 model
14
+ layout_model = LayoutLMv3ForTokenClassification.from_pretrained("omarelsayeed/LayoutReader90Small")
15
 
16
  MAX_LEN = 100
17
  CLS_TOKEN_ID = 0