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