florentgbelidji
/

layoutlmv2-base-uncased

Model card Files Files and versions

florentgbelidji HF Staff commited on Sep 22, 2022

Commit

051d587

·

1 Parent(s): 7573456

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -5,7 +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/cu102/torch1.10/index.html')
 class EndpointHandler():

 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')
 class EndpointHandler():