Transformers How to use spandan21/Image-Segmentation-Model with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import Unet
model = Unet.from_pretrained("spandan21/Image-Segmentation-Model", device_map="auto") 
