Update processing_starvector.py

processing_starvector.py

@@ -64,4 +64,14 @@ class SimpleStarVectorProcessor(ProcessorMixin):
         if text is not None:
             text_inputs = self.tokenizer(text, **kwargs)
         return BatchFeature(data={**text_inputs, **image_inputs})
+
+    def _pad_to_square(self, img):
+        # Calculate padding to make the image square
+        width, height = img.size
+        max_dim = max(width, height)
+        padding = [(max_dim - width) // 2, (max_dim - height) // 2]
+        padding += [max_dim - width - padding[0], max_dim - height - padding[1]]
+        return pad(img, padding, fill=255)  # Assuming white padding
+
+
 AutoProcessor.register(SimpleStarVectorProcessor, SimpleStarVectorProcessor)