gaur3009 commited on
Commit
dfa09bc
·
verified ·
1 Parent(s): 7cc48b2

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 # Import U²-Net
8
 
9
  # Load U²-Net model
10
- model_path = "cloth-segmentation/models/u2net.pth" # Ensure this path is correct
11
  model = U2NET(3, 1)
12
  model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
13
  model.eval()
 
7
  from cloth_segmentation.networks.u2net import U2NET # Import U²-Net
8
 
9
  # Load U²-Net model
10
+ model_path = "cloth_segmentation/networks/u2net.pth" # Ensure this path is correct
11
  model = U2NET(3, 1)
12
  model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
13
  model.eval()