update: app.py

third_party/arcface/mouth_net.py

@@ -65,7 +65,7 @@ class MouthNet(nn.Module):
         torch.save(self.backbone.state_dict(), path)
 
     def load_backbone(self, path: str):
-        self.backbone.load_state_dict(torch.load(path))
+        self.backbone.load_state_dict(torch.load(path, map_location='cpu'))
 
 
 if __name__ == "__main__":