Update app.py

app.py

@@ -40,9 +40,9 @@ def run(*args):
         id_image = resize_numpy_image_long(id_image, 1024)
         id_embeddings = pipeline.get_id_embedding(id_image)
         # for supp_id_image in supp_images:
-        if supp_id_image is not None:
-            supp_id_image = resize_numpy_image_long(supp_id_image, 1024)
-            supp_id_embeddings = pipeline.get_id_embedding(supp_id_image)
+        if supp_images is not None:
+            sup_images = resize_numpy_image_long(supp_images, 1024)
+            supp_id_embeddings = pipeline.get_id_embedding(supp_images)
             id_embeddings = torch.cat(
                 (id_embeddings, supp_id_embeddings if id_mix else supp_id_embeddings[:, :5]), dim=1
             )