Transformers How to use Subh775/Dis-Seg-Former with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("image-segmentation", model="Subh775/Dis-Seg-Former") # Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("Subh775/Dis-Seg-Former", device_map="auto") 
