| | --- |
| | language: |
| | - es |
| | - en |
| | pipeline_tag: image-classification |
| | tags: |
| | - image-classification |
| | widget: |
| | - src: https://upserve.com/media/sites/2/Bill-from-Mezcalero-in-Washington-D.C.-photo-by-Alfredo-Solis-1-e1507226752437.jpg |
| | example_title: receipt |
| | - src: https://templates.invoicehome.com/invoice-template-us-neat-750px.png |
| | example_title: invoice |
| | --- |
| | |
| | **InvoiceReceiptClassifier** is a fine-tuned LayoutLMv2 model that classifies a document to an invoice or receipt. |
| |
|
| | ## Quick start: using the raw model |
| |
|
| | ```python |
| | from transformers import ( |
| | AutoModelForSequenceClassification, |
| | AutoProcessor, |
| | ) |
| | from PIL import Image |
| | from urllib.request import urlopen |
| | |
| | |
| | model = AutoModelForSequenceClassification.from_pretrained("fedihch/InvoiceReceiptClassifier") |
| | processor = AutoProcessor.from_pretrained("fedihch/InvoiceReceiptClassifier") |
| | |
| | input_img_url = "https://upload.wikimedia.org/wikipedia/commons/thumb/0/0b/ReceiptSwiss.jpg/1024px-ReceiptSwiss.jpg" |
| | with urlopen(input_img_url) as testImage: |
| | input_img = Image.open(testImage).convert("RGB") |
| | |
| | encoded_inputs = processor(input_img, padding="max_length", return_tensors="pt") |
| | outputs = model(**encoded_inputs) |
| | logits = outputs.logits |
| | predicted_class_idx = logits.argmax(-1).item() |
| | id2label = {0: "invoice", 1: "receipt"} |
| | print(id2label[predicted_class_idx]) |
| | ``` |
| |
|
| |
|
