diff --git a/app/service/image2sketch/server.py b/app/service/image2sketch/server.py index ebd363e..82f1843 100644 --- a/app/service/image2sketch/server.py +++ b/app/service/image2sketch/server.py @@ -52,8 +52,7 @@ class Image2SketchServer: self.data['B'] = style_img self.data['B'] = self.data['B'].unsqueeze(0).to(device) A, self.width, self.height = self.get_image(self.image_url) - self.data['A'] = transform(A) - self.data['A'] = self.data['A'].unsqueeze(0).to(device) + def get_result(self): self.model.set_input(self.data)