From 5a5bb07f3bb4ac5a39762ce14beaed095654dbc0 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Mon, 30 Sep 2024 11:15:43 +0800 Subject: [PATCH] =?UTF-8?q?feat=20=20image2sketch=20=E5=8F=98=E6=9B=B4?= =?UTF-8?q?=E6=A8=A1=E5=9E=8B=20fix?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/image2sketch_2/server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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()