Update app.py

app.py

@@ -39,13 +39,13 @@ def run(*args):
     if id_image is not None:
         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)
-                id_embeddings = torch.cat(
-                    (id_embeddings, supp_id_embeddings if id_mix else supp_id_embeddings[:, :5]), dim=1
-                )
+        # 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)
+            id_embeddings = torch.cat(
+                (id_embeddings, supp_id_embeddings if id_mix else supp_id_embeddings[:, :5]), dim=1
+            )
     else:
         id_embeddings = None