omarelsayeed
commited on
Commit
•
d94e271
1
Parent(s):
28aab42
Update app.py
Browse files
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)
|