GPU task aborted fix attempt (#12)

- GPU task aborted fix attempt (7ce0ef088e28685cdd111f7e0979326fe64749a8)


Co-authored-by: Charles Bensimon <[email protected]>

app.py

@@ -50,6 +50,8 @@ ip_model_plus = IPAdapterFaceIDPlus(pipe, image_encoder_path, ip_plus_ckpt, devi
 app = FaceAnalysis(name="buffalo_l", providers=['CPUExecutionProvider'])
 app.prepare(ctx_id=0, det_size=(640, 640))
 
+cv2.setNumThreads(1)
+
 @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)):
     faceid_all_embeds = []