enpaiva commited on
Commit
6e49702
·
verified ·
1 Parent(s): 609bb57

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_custom":
23
  {"cfgfile": "configs/dfine/custom/dfine_hgnetv2_n_custom.yml",
24
- "classinfofile": "configs/custom_info.yml",
25
  "weights": "https://github.com/EnPaiva93/storage/raw/refs/heads/main/custom_model.pth"},
26
  "dfine_n_coco":
27
  {"cfgfile": "configs/dfine/dfine_hgnetv2_n_coco.yml",
 
21
  model_configs = {
22
  "dfine_n_custom":
23
  {"cfgfile": "configs/dfine/custom/dfine_hgnetv2_n_custom.yml",
24
+ "classinfofile": "configs/custom_info_coco.yml",
25
  "weights": "https://github.com/EnPaiva93/storage/raw/refs/heads/main/custom_model.pth"},
26
  "dfine_n_coco":
27
  {"cfgfile": "configs/dfine/dfine_hgnetv2_n_coco.yml",