How to use vikp/surya_layout2 with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, SegformerForRegressionMask processor = AutoImageProcessor.from_pretrained("vikp/surya_layout2") model = SegformerForRegressionMask.from_pretrained("vikp/surya_layout2", device_map="auto")
Layout model for surya.
Files info
# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, SegformerForRegressionMask processor = AutoImageProcessor.from_pretrained("vikp/surya_layout2") model = SegformerForRegressionMask.from_pretrained("vikp/surya_layout2", device_map="auto")