Segizu commited on
Commit
057de95
·
1 Parent(s): 389fb53
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -4,6 +4,14 @@ import cv2
4
  import numpy as np
5
  import os
6
  from PIL import Image
 
 
 
 
 
 
 
 
7
 
8
  # 📂 Ruta donde están las imágenes
9
  IMAGE_DIRECTORY = "dataset_faces/"
 
4
  import numpy as np
5
  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.")
12
+ else:
13
+ st.write("CUDA is not available. Using CPU for face recognition.")
14
+
15
 
16
  # 📂 Ruta donde están las imágenes
17
  IMAGE_DIRECTORY = "dataset_faces/"