How to use gdurkin/segformer-b0-finetuned-segments-floods-S2-pseudoRGBv1 with Transformers:
# Load model directly from transformers import AutoImageProcessor, SegformerForSemanticSegmentation processor = AutoImageProcessor.from_pretrained("gdurkin/segformer-b0-finetuned-segments-floods-S2-pseudoRGBv1") model = SegformerForSemanticSegmentation.from_pretrained("gdurkin/segformer-b0-finetuned-segments-floods-S2-pseudoRGBv1", device_map="auto")
