Update app.py

app.py

@@ -39,7 +39,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
     vae=vae,
     feature_extractor=safety_feature_extractor,
     safety_checker=safety_checker
-)
+).to(device)
 
 #pipe.load_lora_weights("h94/IP-Adapter-FaceID", weight_name="ip-adapter-faceid-plusv2_sd15_lora.safetensors")
 #pipe.fuse_lora()
@@ -47,12 +47,11 @@ pipe = StableDiffusionPipeline.from_pretrained(
 ip_model = IPAdapterFaceID(pipe, ip_ckpt, device)
 ip_model_plus = IPAdapterFaceIDPlus(pipe, image_encoder_path, ip_plus_ckpt, device)
 
+app = FaceAnalysis(name="buffalo_l", providers=['CPUExecutionProvider'])
+app.prepare(ctx_id=0, det_size=(640, 640))
+
 @spaces.GPU(enable_queue=True)
 def generate_image(images, prompt, negative_prompt, preserve_face_structure, face_strength, likeness_strength, nfaa_negative_prompt, progress=gr.Progress(track_tqdm=True)):
-    pipe.to(device)
-    app = FaceAnalysis(name="buffalo_l", providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
-    app.prepare(ctx_id=0, det_size=(640, 640))
-    
     faceid_all_embeds = []
     first_iteration = True
     for image in images: