Spaces:

hysts
/

CelebAMask-HQ-Face-Parsing

Running on Zero

hysts HF Staff commited on May 28, 2024

Commit

39e4135

1 Parent(s): 49d2e89

Update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ transform = T.Compose(
     ]
 )
-path = hf_hub_download("hysts/CelebAMask-HQ-Face-Parsing", "models/model.pth")
 state_dict = torch.load(path, map_location="cpu")
 model = unet()
 model.load_state_dict(state_dict)

     ]
 )
+path = hf_hub_download("public-data/CelebAMask-HQ-Face-Parsing", "models/model.pth")
 state_dict = torch.load(path, map_location="cpu")
 model = unet()
 model.load_state_dict(state_dict)