florentgbelidji
/

layoutlmv2-base-uncased

Model card Files Files and versions

florentgbelidji HF Staff commited on Sep 22, 2022

Commit

02e4833

·

1 Parent(s): 5585b0a

Update handler.py

Files changed (1) hide show

handler.py +1 -0

handler.py CHANGED Viewed

@@ -5,6 +5,7 @@ from PIL import Image, ImageDraw, ImageFont
 from transformers import LayoutLMv2Processor, LayoutLMv2ForTokenClassification
 from transformers import pipeline, AutoTokenizer
 os.system('pip install -q detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html')
 import detectron2

 from transformers import LayoutLMv2Processor, LayoutLMv2ForTokenClassification
 from transformers import pipeline, AutoTokenizer
+os.system('pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI')
 os.system('pip install -q detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html')
 import detectron2