Spaces:

gaur3009
/

detecting_dress

Running

gaur3009 commited on about 1 month ago

Commit

c274843

verified ·

1 Parent(s): 3eff549

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("cloth_segementation/networks/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_segmentation/networks/u2net.pth", map_location=torch.device('cpu')))
 model.eval()
 # Preprocessing