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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -119,7 +119,7 @@ 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)
 
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_EVA02_Scaleup10m.pth')
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)