Spaces:
Runtime error
Runtime error
from transformers import LayoutLMv2Processor, LayoutLMv2ForTokenClassification | |
from PIL import Image | |
processor = LayoutLMv2Processor.from_pretrained("microsoft/layoutlmv2-base-uncased", revision="no_ocr") | |
model = LayoutLMv2ForTokenClassification.from_pretrained("microsoft/layoutlmv2-base-uncased") | |
image = Image.open("Labels-New\OriginalImages_cropped_ocr_results\IMG_1693.bmp").convert("RGB") | |
words = ["hello", "world"] | |
boxes = [[1, 2, 3, 4], [5, 6, 7, 8]] # make sure to normalize your bounding boxes | |
word_labels = [0, 1] | |
encoding = processor(image, words, boxes=boxes, word_labels=word_labels, return_tensors="pt") | |
outputs = model(**encoding) | |
loss = outputs.loss | |
logits = outputs.logits |