diff --git a/app/service/image2sketch_2/server.py b/app/service/image2sketch_2/server.py index 93c9574..41c0278 100644 --- a/app/service/image2sketch_2/server.py +++ b/app/service/image2sketch_2/server.py @@ -92,7 +92,7 @@ class Generator(nn.Module): model1 = Generator(3, 1, 3) -model1.load_state_dict(torch.load('service/image2sketch_2/model.pth', map_location=torch.device('cpu'))) +model1.load_state_dict(torch.load('app/service/image2sketch_2/model.pth', map_location=torch.device('cpu'))) model1.eval()