mich123geb commited on
Commit
2ae66d0
Β·
verified Β·
1 Parent(s): be2061f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ if not MODEL_PATH.exists():
21
  # ──────────────────────────────────────────────
22
  # 2. Face detection setup
23
  # ──────────────────────────────────────────────
24
- fa = face_alignment.FaceAlignment("2D", flip_input=False)
25
 
26
 
27
  def crop_face(image: Image.Image) -> Image.Image:
 
21
  # ──────────────────────────────────────────────
22
  # 2. Face detection setup
23
  # ──────────────────────────────────────────────
24
+ fa = face_alignment.FaceAlignment("2D", flip_input=False, device='cpu')
25
 
26
 
27
  def crop_face(image: Image.Image) -> Image.Image: