omarelsayeed commited on
Commit
d94e271
1 Parent(s): 28aab42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -10,8 +10,12 @@ import torch
10
  from transformers import LayoutLMv3ForTokenClassification
11
  from transformers import AutoProcessor
12
  from transformers import AutoModelForTokenClassification
 
 
 
 
 
13
 
14
- finetuned_fully = LayoutLMv3ForTokenClassification.from_pretrained("omarelsayeed/YARAB_FOK_ELDE2A")
15
 
16
  processor = AutoProcessor.from_pretrained("microsoft/layoutlmv3-base",
17
  apply_ocr=False)
 
10
  from transformers import LayoutLMv3ForTokenClassification
11
  from transformers import AutoProcessor
12
  from transformers import AutoModelForTokenClassification
13
+ quant=True
14
+ if quant:
15
+ finetuned_fully = LayoutLMv3ForTokenClassification.from_pretrained("omarelsayeed/ArabicLayoutReader_4bit")
16
+ else:
17
+ finetuned_fully = LayoutLMv3ForTokenClassification.from_pretrained("omarelsayeed/YARAB_FOK_ELDE2A")
18
 
 
19
 
20
  processor = AutoProcessor.from_pretrained("microsoft/layoutlmv3-base",
21
  apply_ocr=False)