Spaces:

XavierJiezou
/

face-mogle

Runtime error

XavierJiezou commited on Apr 12

Commit

337046e

verified ·

1 Parent(s): 41c83f6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -628,7 +628,7 @@ class ImageGenerator:
             model_path="https://github.com/FacePerceiver/facer/releases/download/models-v1/face_parsing.farl.celebm.main_ema_181500_jit.pt",
         )
         self.segface = SegFaceCeleb(512, "convnext_base").to(self.args.device)
-        checkpoint = torch.hub.load_state_dict_from_url("https://huggingface.co/kartiknarayan/SegFace/resolve/main/convnext_celeba_512/model_299.pt")
         self.segface.load_state_dict(checkpoint["state_dict_backbone"])
         self.segface.eval()
         self.segface_transforms = torchvision.transforms.Compose(

             model_path="https://github.com/FacePerceiver/facer/releases/download/models-v1/face_parsing.farl.celebm.main_ema_181500_jit.pt",
         )
         self.segface = SegFaceCeleb(512, "convnext_base").to(self.args.device)
+        checkpoint = torch.hub.load_state_dict_from_url("https://huggingface.co/kartiknarayan/SegFace/resolve/main/convnext_celeba_512/model_299.pt",map_location="cpu")
         self.segface.load_state_dict(checkpoint["state_dict_backbone"])
         self.segface.eval()
         self.segface_transforms = torchvision.transforms.Compose(