Update app.py
Browse files
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/
|
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",
|