enpaiva commited on
Commit
cb1aed1
·
verified ·
1 Parent(s): 03e284f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ from src.core import YAMLConfig
21
  model_configs = {
22
  "dfine_n_coco":
23
  {"cfgfile": "configs/dfine/dfine_hgnetv2_n_coco.yml",
24
- "classinfofile": "configs/coco.yml",
25
  "weights": "https://github.com/Peterande/storage/releases/download/dfinev1.0/dfine_n_coco.pth"},
26
  "dfine_s_coco":
27
  {"cfgfile": "configs/dfine/dfine_hgnetv2_s_coco.yml",
 
21
  model_configs = {
22
  "dfine_n_coco":
23
  {"cfgfile": "configs/dfine/dfine_hgnetv2_n_coco.yml",
24
+ "classinfofile": "configs/custom_info.yml",
25
  "weights": "https://github.com/Peterande/storage/releases/download/dfinev1.0/dfine_n_coco.pth"},
26
  "dfine_s_coco":
27
  {"cfgfile": "configs/dfine/dfine_hgnetv2_s_coco.yml",