Update inference.py

inference.py

@@ -15,7 +15,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model.to(device)
 
 
-def prediction(image):
+def prediction(Image.open(image)):
     boxes, words = OCR(image)
     encoding = processor(image, words, boxes=boxes, return_offsets_mapping=True, return_tensors="pt", truncation=True)
     offset_mapping = encoding.pop('offset_mapping')