Transformers How to use vikp/surya_layout2 with Transformers:
# 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") 
