primerz commited on
Commit
8f11caf
·
verified ·
1 Parent(s): 0a65708

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +2 -2
models.py CHANGED
@@ -73,7 +73,7 @@ def load_face_analysis():
73
  # Download antelopev2 model files
74
  print(" Downloading antelopev2 model files...")
75
  snapshot_download(
76
- repo_id=FACE_DETECTION_CONFIG['download_repo'],
77
  local_dir=FACE_DETECTION_CONFIG['local_dir']
78
  )
79
  print(" [OK] Antelopev2 downloaded")
@@ -83,7 +83,7 @@ def load_face_analysis():
83
 
84
  print(f" Initializing FaceAnalysis with providers: {providers}")
85
  face_app = FaceAnalysis(
86
- name=FACE_DETECTION_CONFIG['model_name'],
87
  root=FACE_DETECTION_CONFIG['root'],
88
  providers=providers
89
  )
 
73
  # Download antelopev2 model files
74
  print(" Downloading antelopev2 model files...")
75
  snapshot_download(
76
+ repo_id=FACE_DETECTION_CONFIG['repo_id'],
77
  local_dir=FACE_DETECTION_CONFIG['local_dir']
78
  )
79
  print(" [OK] Antelopev2 downloaded")
 
83
 
84
  print(f" Initializing FaceAnalysis with providers: {providers}")
85
  face_app = FaceAnalysis(
86
+ name=FACE_DETECTION_CONFIG['name'],
87
  root=FACE_DETECTION_CONFIG['root'],
88
  providers=providers
89
  )