Spaces:
Paused
Paused
Update app.py
Browse files
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 |
-
|
122 |
checkpoints_eva = torch.load('GLEE/GLEE_{}.pth'.format(args.version))
|
123 |
-
cfg_eva02.merge_from_file(
|
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()
|