gaur3009 commited on
Commit
6e5e70e
·
verified ·
1 Parent(s): 0bda8ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from torchvision import transforms
7
  from cloth_segmentation.networks.u2net import U2NET
8
 
9
  # Load U²-Net model
10
- model_path = "/kaggle/input/tygsgg/cloth_segmentation/networks/u2net.pth"
11
  model = U2NET(3, 1)
12
  state_dict = torch.load(model_path, map_location=torch.device('cpu'))
13
  state_dict = {k.replace('module.', ''): v for k, v in state_dict.items()}
 
7
  from cloth_segmentation.networks.u2net import U2NET
8
 
9
  # Load U²-Net model
10
+ model_path = "cloth_segmentation/networks/u2net.pth"
11
  model = U2NET(3, 1)
12
  state_dict = torch.load(model_path, map_location=torch.device('cpu'))
13
  state_dict = {k.replace('module.', ''): v for k, v in state_dict.items()}