How to use wieheistdu/segment_5ep with Transformers:
# Load model directly from transformers import AutoImageProcessor, SegformerForSemanticSegmentation processor = AutoImageProcessor.from_pretrained("wieheistdu/segment_5ep") model = SegformerForSemanticSegmentation.from_pretrained("wieheistdu/segment_5ep", device_map="auto")
