omarelsayeed commited on
Commit
db9f08e
1 Parent(s): 8645db9

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/test_3")
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/trial_simple_random_f")
15
 
16
  MAX_LEN = 100
17
  CLS_TOKEN_ID = 0