Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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('
|
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)
|