zdou0830 commited on
Commit
3c3d474
1 Parent(s): d1f9d4c
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -65,6 +65,7 @@ glip_demo = GLIPDemo(
65
 
66
  config_file = "configs/pretrain_new/desco_fiber.yaml"
67
  weight_file = "MODEL/desco_fiber_base.pth"
 
68
  cfg.merge_from_list(["MODEL.WEIGHT", weight_file])
69
  cfg.merge_from_list(["MODEL.DEVICE", "cuda"])
70
  fiber_demo = GLIPDemo(
 
65
 
66
  config_file = "configs/pretrain_new/desco_fiber.yaml"
67
  weight_file = "MODEL/desco_fiber_base.pth"
68
+ cfg.merge_from_file(config_file)
69
  cfg.merge_from_list(["MODEL.WEIGHT", weight_file])
70
  cfg.merge_from_list(["MODEL.DEVICE", "cuda"])
71
  fiber_demo = GLIPDemo(