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