hysts HF Staff commited on
Commit
39e4135
·
1 Parent(s): 49d2e89
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ transform = T.Compose(
29
  ]
30
  )
31
 
32
- path = hf_hub_download("hysts/CelebAMask-HQ-Face-Parsing", "models/model.pth")
33
  state_dict = torch.load(path, map_location="cpu")
34
  model = unet()
35
  model.load_state_dict(state_dict)
 
29
  ]
30
  )
31
 
32
+ path = hf_hub_download("public-data/CelebAMask-HQ-Face-Parsing", "models/model.pth")
33
  state_dict = torch.load(path, map_location="cpu")
34
  model = unet()
35
  model.load_state_dict(state_dict)