How to use capofwesh20/my-segmentation-model with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, SegformerForSemanticSegmentation processor = AutoImageProcessor.from_pretrained("capofwesh20/my-segmentation-model") model = SegformerForSemanticSegmentation.from_pretrained("capofwesh20/my-segmentation-model", device_map="auto")
