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

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +4 -4
config.py CHANGED
@@ -36,12 +36,12 @@ TRIGGER_WORD = "p1x3l4rt, pixel art"
36
 
37
  # Face detection configuration
38
  FACE_DETECTION_CONFIG = {
39
- "model_name": "antelopev2",
40
  "det_size": (640, 640),
41
  "ctx_id": 0,
42
- "download_repo": "DIAMONIK7777/antelopev2",
43
- "local_dir": "./models/antelopev2",
44
- "root": "./models",
45
  "providers": ['CPUExecutionProvider'],
46
  }
47
 
 
36
 
37
  # Face detection configuration
38
  FACE_DETECTION_CONFIG = {
39
+ "name": "antelopev2",
40
  "det_size": (640, 640),
41
  "ctx_id": 0,
42
+ "repo_id": "DIAMONIK7777/antelopev2",
43
+ "local_dir": "/data/models/antelopev2",
44
+ "root": "/data",
45
  "providers": ['CPUExecutionProvider'],
46
  }
47