Update README.md

README.md

@@ -20,16 +20,14 @@ from transformers import (
     AutoModelForSequenceClassification,
     AutoProcessor,
 )
-model = AutoModelForSequenceClassification.from_pretrained("fedihch/InvoiceReceiptClassifier")
+from PIL import Image
 
+model = AutoModelForSequenceClassification.from_pretrained("fedihch/InvoiceReceiptClassifier")
 processor = AutoProcessor.from_pretrained("fedihch/InvoiceReceiptClassifier")
-```
-```python
-from PIL import Image
-input_img = Image.open("*****.jpg")
-w, h = input_img.size
-input_img = input_img.convert("RGB").resize((int(w * 600 / h), 600))
-encoded_inputs = processor(input_img, return_tensors="pt")
+
+
+input_img = Image.open("https://upload.wikimedia.org/wikipedia/commons/thumb/0/0b/ReceiptSwiss.jpg/1024px-ReceiptSwiss.jpg")
+encoded_inputs = processor(input_img, padding="max_length", return_tensors="pt")
 for k, v in encoded_inputs.items():
     encoded_inputs[k] = v.to(model.device)
 outputs = model(**encoded_inputs)