Views
No views yet
1from models import ISNetDIS
2import torch
3
4# Load model
5model = ISNetDIS()
6state_dict = torch.load("supplyswap_isnet.pth", map_location="cpu")
7model.load_state_dict(state_dict)
8model.eval()
9
10# Use for inference
11# See the Spaces demo for complete usage examplessupplyswap_isnet.pth: Pre-trained model weightsmodels/: Model architecture implementation
isnet.py: ISNet model implementation__init__.py: Model imports