Hugging Face's logo

Spaces:
omarelsayeed
/
RT-Detr-ArabicLayoutAnalysisR
Sleeping

App Files Community
omarelsayeed commited on
Commit
0f127fa
·
verified ·
1 Parent(s): d94e271

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ 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
 
 
12
  from transformers import AutoModelForTokenClassification
13
  quant=True
14
  if quant:
15
+ finetuned_fully = LayoutLMv3ForTokenClassification.from_pretrained("omarelsayeed/ArabicLayoutReader_4bit" , load_in_4bit=True)
16
  else:
17
  finetuned_fully = LayoutLMv3ForTokenClassification.from_pretrained("omarelsayeed/YARAB_FOK_ELDE2A")
18