Hugging Face's logo

Spaces:
Segizu
/
face_recognition
Build error

App Files Community
Segizu commited on
Commit
e52597d
·
1 Parent(s): 4eb2e35

as

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -6,6 +6,9 @@ import os
6
  from PIL import Image
7
  import dlib
8
 
 
 
 
9
  # Check if CUDA is available
10
  if dlib.DLIB_USE_CUDA:
11
  st.write("CUDA is available. Using GPU for face recognition.")
 
6
  from PIL import Image
7
  import dlib
8
 
9
+
10
+ device = "cuda"
11
+
12
  # Check if CUDA is available
13
  if dlib.DLIB_USE_CUDA:
14
  st.write("CUDA is available. Using GPU for face recognition.")