Update app.py

app.py

@@ -33,18 +33,15 @@ pipe = StableDiffusionPipeline.from_pretrained(
 
 ip_model = IPAdapterFaceID(pipe, ip_ckpt, device)
 
-def generate_faceid_embeddings(image):
-    #image = cv2.imread("person.jpg")
-    faces = app.get(image)
-    faceid_embeds = torch.from_numpy(faces[0].normed_embedding).unsqueeze(0)
-    return faceid_embeds
-
 @spaces.GPU
 def generate_image(image, prompt, negative_prompt):
+    pipe.to(device)
+    
     app = FaceAnalysis(name="buffalo_l", providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
     app.prepare(ctx_id=0, det_size=(640, 640))
-    pipe.to(device)
-    faceid_embeds = generate_faceid_embeddings(image)
+    faces = app.get(image)
+    faceid_embeds = torch.from_numpy(faces[0].normed_embedding).unsqueeze(0)
+    
     images = ip_model.generate(
         prompt=prompt, negative_prompt=negative_prompt, faceid_embeds=faceid_embeds, width=512, height=512, num_inference_steps=30
     )