Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -53,7 +53,7 @@ app.prepare(ctx_id=0, det_size=(640, 640))
|
|
53 |
|
54 |
cv2.setNumThreads(1)
|
55 |
|
56 |
-
@spaces.
|
57 |
def generate_image(images, prompt, negative_prompt, preserve_face_structure, face_strength, likeness_strength, nfaa_negative_prompt, progress=gr.Progress(track_tqdm=True)):
|
58 |
faceid_all_embeds = []
|
59 |
first_iteration = True
|
|
|
53 |
|
54 |
cv2.setNumThreads(1)
|
55 |
|
56 |
+
@spaces.GPU(enable_queue=True) # Changed to GPU to allow flexibility between CPU/GPU
|
57 |
def generate_image(images, prompt, negative_prompt, preserve_face_structure, face_strength, likeness_strength, nfaa_negative_prompt, progress=gr.Progress(track_tqdm=True)):
|
58 |
faceid_all_embeds = []
|
59 |
first_iteration = True
|