How to use HorcruxNo13/segformer-b0-finetuned-segments-toolwear with Transformers:
# Load model directly from transformers import AutoImageProcessor, SegformerForSemanticSegmentation processor = AutoImageProcessor.from_pretrained("HorcruxNo13/segformer-b0-finetuned-segments-toolwear") model = SegformerForSemanticSegmentation.from_pretrained("HorcruxNo13/segformer-b0-finetuned-segments-toolwear", device_map="auto")
How to fix it?
