Download checkpoint from HF hub in OcrReorderPipeline

inference.py

@@ -29,13 +29,13 @@ class OcrReorderPipeline(Pipeline):
         self.projection.to(self.device)
 
     def _sanitize_parameters(self, **kwargs):
-        # extract the pipeline 'inputs' (base64 image) and custom args
-        image = kwargs.pop("inputs", None)
-        words = kwargs.pop("words", None)
-        boxes = kwargs.pop("boxes", None)
-        return {"image": image, "words": words, "boxes": boxes}, {}, {}
+        # Extract only the custom args for preprocess; 'inputs' (the image) is passed positionally
+        words = kwargs.get("words", None)
+        boxes = kwargs.get("boxes", None)
+        return {"words": words, "boxes": boxes}, {}, {}
 
     def preprocess(self, image, words, boxes):
+        # 'image' comes from the positional 'inputs' argument
         data = base64.b64decode(image)
         img  = Image.open(BytesIO(data)).convert("RGB")
         return self.feature_extractor(