Junfeng5 commited on
Commit
4111de0
1 Parent(s): 281ad49

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -118,9 +118,9 @@ GLEEmodel_swin.eval()
118
  cfg_eva02 = get_cfg()
119
  add_deeplab_config(cfg_eva02)
120
  add_glee_config(cfg_eva02)
121
- conf_files_swin = 'GLEE/configs/EVA02.yaml'
122
  checkpoints_eva = torch.load('GLEE/GLEE_{}.pth'.format(args.version))
123
- cfg_eva02.merge_from_file(conf_files_swin)
124
  GLEEmodel_eva02 = GLEE_Model(cfg_eva02, None, device, None, True).to(device)
125
  GLEEmodel_eva02.load_state_dict(checkpoints_eva, strict=False)
126
  GLEEmodel_eva02.eval()
 
118
  cfg_eva02 = get_cfg()
119
  add_deeplab_config(cfg_eva02)
120
  add_glee_config(cfg_eva02)
121
+ conf_files_eva02 = 'GLEE/configs/EVA02.yaml'
122
  checkpoints_eva = torch.load('GLEE/GLEE_{}.pth'.format(args.version))
123
+ cfg_eva02.merge_from_file(conf_files_eva02)
124
  GLEEmodel_eva02 = GLEE_Model(cfg_eva02, None, device, None, True).to(device)
125
  GLEEmodel_eva02.load_state_dict(checkpoints_eva, strict=False)
126
  GLEEmodel_eva02.eval()