changes

3 years ago · 0e8901749e
1 changed files with 2 additions and 1 deletions
--- a/animegan.py
+++ b/animegan.py
@ -27,8 +27,9 @@ class Animegan(Operator):
        self.tfms = transforms.Compose([
                                        transforms.ToTensor()
                                    ])
    @arg(1, to_image_color('RGB') )
    @arg(1, to_image_color('RGB'))
    def __call__(self, image):
        img = self.tfms(image).unsqueeze(0)
        print(img.shape)
        styled_image = self.model(img)
        return Image(styled_image, 'RGB')