Update app.py (#4)

- Update app.py (b8a9073b6b76c698ad1afd8f04a610af97cbc9d1)


Co-authored-by: Yanze Wu <[email protected]>

app.py

@@ -107,7 +107,7 @@ def run(*args):
         ims2 = []
         for idx, (b, im1, im2, cond_weight) in enumerate(zip(*inps)):
             if idx > 1:
-                if im1 is not None or im2 is not None:
+                if b != 'Nothing' and (im1 is not None or im2 is not None):
                     if im1 is not None:
                         h, w, _ = im1.shape
                     else:
@@ -119,10 +119,11 @@ def run(*args):
                 ims1.append(im1)
                 ims2.append(im2)
                 continue
-            if im1 is not None:
-                im1 = cv2.resize(im1, (w, h), interpolation=cv2.INTER_CUBIC)
-            if im2 is not None:
-                im2 = cv2.resize(im2, (w, h), interpolation=cv2.INTER_CUBIC)
+            if b != 'Nothing':
+                if im1 is not None:
+                    im1 = cv2.resize(im1, (w, h), interpolation=cv2.INTER_CUBIC)
+                if im2 is not None:
+                    im2 = cv2.resize(im2, (w, h), interpolation=cv2.INTER_CUBIC)
             ims1.append(im1)
             ims2.append(im2)