Update app.py

app.py

@@ -106,13 +106,13 @@ def run(*args):
         ims1 = []
         ims2 = []
         for idx, (b, im1, im2, cond_weight) in enumerate(zip(*inps)):
-            # if idx > 1:
-            if b != 'Nothing' and (im1 is not None or im2 is not None):
-                if im1 is not None:
-                    h, w, _ = im1.shape
-                else:
-                    h, w, _ = im2.shape
-                break
+            if idx > 0:
+                if b != 'Nothing' and (im1 is not None or im2 is not None):
+                    if im1 is not None:
+                        h, w, _ = im1.shape
+                    else:
+                        h, w, _ = im2.shape
+                # break
         # resize all the images to the same size
         for idx, (b, im1, im2, cond_weight) in enumerate(zip(*inps)):
             if idx == 0: