fix

1 files changed, 3 insertions, 2 deletions

diff --git a/live-mogrify.py b/live-mogrify.py
index 0725753..f4200a3 100644
--- a/live-mogrify.py
+++ b/live-mogrify.py
@@ -158,7 +158,7 @@ def list_sequences(module):
 
 import torchvision.transforms as transforms
 
-def get_transform(opt):
+def get_transform(opt={}):
   transform_list = []
   # if opt.resize_or_crop == 'resize_and_crop':
   #   osize = [opt.loadSize, opt.loadSize]
@@ -190,8 +190,9 @@ def load_frame(opt, index):
   if not os.path.exists(A_path):
     print("path doesn't exist: {}".format(A_path))
     return None
+  transform = get_transform()
   A = Image.open(A_path).convert('RGB')
-  A = self.transform(A_img)
+  A = transform(A_img)
   # if self.opt.which_direction == 'BtoA':
   #     input_nc = self.opt.output_nc
   # else: