xichen98cn commited on
Commit
13928bb
·
verified ·
1 Parent(s): 13fb3dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ import json
26
  setup_logger()
27
  logger = setup_logger(name="frozenseg")
28
  cfg = get_cfg()
29
- cfg.MODEL.DEVICE='cuda'
30
  add_maskformer2_config(cfg)
31
  add_frozenseg_config(cfg)
32
  cfg.merge_from_file("configs/coco/frozenseg/convnext_large_eval_ade20k.yaml")
 
26
  setup_logger()
27
  logger = setup_logger(name="frozenseg")
28
  cfg = get_cfg()
29
+ cfg.MODEL.DEVICE='cpu'
30
  add_maskformer2_config(cfg)
31
  add_frozenseg_config(cfg)
32
  cfg.merge_from_file("configs/coco/frozenseg/convnext_large_eval_ade20k.yaml")