enpaiva commited on
Commit
7dfa8d2
·
verified ·
1 Parent(s): cb1aed1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -22,6 +22,10 @@ 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",
 
22
  "dfine_n_coco":
23
  {"cfgfile": "configs/dfine/dfine_hgnetv2_n_coco.yml",
24
  "classinfofile": "configs/custom_info.yml",
25
+ "weights": "custom_model.pth"},
26
+ "dfine_n_coco":
27
+ {"cfgfile": "configs/dfine/dfine_hgnetv2_n_coco.yml",
28
+ "classinfofile": "configs/coco.yml",
29
  "weights": "https://github.com/Peterande/storage/releases/download/dfinev1.0/dfine_n_coco.pth"},
30
  "dfine_s_coco":
31
  {"cfgfile": "configs/dfine/dfine_hgnetv2_s_coco.yml",