diff --git a/animegan.py b/animegan.py
index 18c3a08..47ed90f 100644
--- 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')