Spaces:

xvjiarui
/

GroupViT

Runtime error

xvjiarui commited on Mar 27, 2022

Commit

79922d7

1 Parent(s): ce5c8b9

new model link

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,8 @@ from segmentation.evaluation import (GROUP_PALETTE, build_seg_demo_pipeline,
 from utils import get_config, load_checkpoint
 os.chdir('GroupViT')
-checkpoint_url = 'https://github.com/xvjiarui/GroupViT-1/releases/download/v1.0.0/group_vit_gcc_yfcc_30e-74d335e6.pth'
 cfg_path = 'configs/group_vit_gcc_yfcc_30e.yml'
 output_dir = 'demo/output'
 device = 'cpu'

 from utils import get_config, load_checkpoint
 os.chdir('GroupViT')
+# checkpoint_url = 'https://github.com/xvjiarui/GroupViT/releases/download/v1.0.0/group_vit_gcc_yfcc_30e-74d335e6.pth'
+checkpoint_url = 'https://github.com/xvjiarui/GroupViT/releases/download/v1.0.0/group_vit_gcc_yfcc_30e-879422e0.pth'
 cfg_path = 'configs/group_vit_gcc_yfcc_30e.yml'
 output_dir = 'demo/output'
 device = 'cpu'