Spaces:

gaur3009
/

detecting_dress

Running

gaur3009 commited on Jun 16

Commit

3eff549

verified ·

1 Parent(s): d93d0a4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ from sklearn.cluster import KMeans
 from cloth_segmentation.networks.u2net import U2NET
 model = U2NET(3, 1)
-model.load_state_dict(torch.load("u2net.pth", map_location=torch.device('cpu')))
 model.eval()
 # Preprocessing

 from cloth_segmentation.networks.u2net import U2NET
 model = U2NET(3, 1)
+model.load_state_dict(torch.load("cloth_segementation/networks/u2net.pth", map_location=torch.device('cpu')))
 model.eval()
 # Preprocessing