From 0e8901749e9dae1020ccb1f73edf81b0aa3ce1cf Mon Sep 17 00:00:00 2001
From: Filip Haltmayer <filip.haltmayer@zilliz.com>
Date: Mon, 4 Apr 2022 13:58:54 -0700
Subject: [PATCH] changes

---
 animegan.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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')