Update app.py

app.py

@@ -53,11 +53,14 @@ def transform_image(face_image):
     style_image_path = "examples/soyjak2.jpg"  # Ensure this path is correct
     style_image = Image.open(style_image_path).resize(desired_size, Image.LANCZOS).convert('RGB')
     style_image_tensor = transforms.ToTensor()(style_image).unsqueeze(0).to("cuda")
+    
+    # Convert the processed face image to tensor and move to GPU
+    processed_face_image_tensor = transforms.ToTensor()(processed_face_image).unsqueeze(0).to("cuda")
 
     # Perform the transformation using the configured pipeline
     image = pipeline(
         prompt="soyjak",
-        ip_adapter_image=[style_image_tensor, transforms.ToTensor()(processed_face_image).unsqueeze(0).to("cuda")],  # Ensure these are tensors
+        ip_adapter_image=[style_image_tensor, processed_face_image_tensor],  # Ensure these are tensors
         negative_prompt="monochrome, lowres, bad anatomy, worst quality, low quality",
         num_inference_steps=30,
         generator=generator,
@@ -80,4 +83,4 @@ demo = gr.Interface(
 )
 
 demo.queue(max_size=20)
-demo.launch()
+demo.launch()