KseniaAI commited on
Commit
0c7e9b0
β€’
1 Parent(s): e83203e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ app.prepare(ctx_id=0, det_size=(640, 640))
53
 
54
  cv2.setNumThreads(1)
55
 
56
- @spaces.Interface(enable_queue=True) # Changed to Interface 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
 
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