pierreguillou commited on
Commit
7c00128
·
1 Parent(s): 00b1f3e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -55,6 +55,9 @@ os.system('python -m pip install --upgrade pip')
55
  model_id_lilt = "pierreguillou/lilt-xlm-roberta-base-finetuned-with-DocLayNet-base-at-paragraphlevel-ml512"
56
  model_id_layoutxlm = "pierreguillou/layout-xlm-base-finetuned-with-DocLayNet-base-at-paragraphlevel-ml512"
57
 
 
 
 
58
  # get device
59
  import torch
60
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
55
  model_id_lilt = "pierreguillou/lilt-xlm-roberta-base-finetuned-with-DocLayNet-base-at-paragraphlevel-ml512"
56
  model_id_layoutxlm = "pierreguillou/layout-xlm-base-finetuned-with-DocLayNet-base-at-paragraphlevel-ml512"
57
 
58
+ # tokenizer for LayoutXLM
59
+ tokenizer_id_layoutxlm = "xlm-roberta-base"
60
+
61
  # get device
62
  import torch
63
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")