Update app.py

app.py

@@ -10,6 +10,7 @@ from pulid.utils import resize_numpy_image_long, seed_everything
 
 import boto3
 from botocore.exceptions import ClientError
+from PIL import Image
 
 torch.set_grad_enabled(False)
 
@@ -93,10 +94,10 @@ def run(bucket_folder, person_name, req_id):
     id_image = image_paths[0]
 
     if image_paths is not None:
-        id_image = resize_numpy_image_long(image_paths[0], 1024)
+        id_image = resize_numpy_image_long(np.array(Image.open(image_paths[0])), 1024)
         id_embeddings = pipeline.get_id_embedding(id_image)
         for i in range(1,len(image_paths)):
-            supp_id_image = resize_numpy_image_long(image_paths[i], 1024)
+            supp_id_image = resize_numpy_image_long(np.array(Image.open(image_paths[i])), 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