Update app.py

app.py

@@ -116,7 +116,7 @@ def run(*args):
         ims1 = []
         ims2 = []
         for idx, (b, im1, im2, cond_weight) in enumerate(zip(*inps)):
-            if idx > 1:
+            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