primerz commited on
Commit
182139b
·
verified ·
1 Parent(s): f3383b1

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +1 -1
models.py CHANGED
@@ -72,7 +72,7 @@ def load_face_analysis():
72
  antelope_download = snapshot_download(repo_id="DIAMONIK7777/antelopev2", local_dir="/data/models/antelopev2")
73
  face_app = FaceAnalysis(name='antelopev2', root='/data', providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
74
  face_app.prepare(ctx_id=0, det_size=(640, 640))
75
- return face_app, True
76
 
77
  except Exception as e:
78
  print(f" [ERROR] Face detection not available: {e}")
 
72
  antelope_download = snapshot_download(repo_id="DIAMONIK7777/antelopev2", local_dir="/data/models/antelopev2")
73
  face_app = FaceAnalysis(name='antelopev2', root='/data', providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
74
  face_app.prepare(ctx_id=0, det_size=(640, 640))
75
+ return face_app, True
76
 
77
  except Exception as e:
78
  print(f" [ERROR] Face detection not available: {e}")