Update app.py

app.py

@@ -6,7 +6,6 @@ import os
 import random
 import spaces
 from PIL import Image
-from diffusers.utils import load_image
 
 hf_token = os.environ.get("HF_TOKEN")
 from diffusers import AutoPipelineForText2Image
@@ -25,13 +24,10 @@ def predict(prompt, ip_adapter_image, ip_adapter_scale=0.5, negative_prompt="",
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
-    # if not center_crop:
-    #     ip_adapter_image = [x.resize((224,224)) for x in ip_adapter_image[0]]
-    # else:
-    #     ip_adapter_image = ip_adapter_image[0]
-    input_id_images = []
-    for img in ip_adapter_image:
-        input_id_images.append(load_image(img))
+    if not center_crop:
+        ip_adapter_image = [x.resize((224,224)) for x in ip_adapter_image[0]]
+    else:
+        ip_adapter_image = ip_adapter_image[0]
 
     generator = torch.Generator(device="cuda").manual_seed(seed)
     pipe.set_ip_adapter_scale([ip_adapter_scale])