Spaces:
Running
on
Zero
Running
on
Zero
Update
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ transform = T.Compose(
|
|
29 |
]
|
30 |
)
|
31 |
|
32 |
-
path = hf_hub_download("
|
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)
|