xvjiarui commited on
Commit
7f4fb32
Β·
1 Parent(s): 5cc72c8

fixed path

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,7 +2,7 @@
2
  import os
3
  import sys
4
  os.system("git clone https://github.com/NVlabs/GroupViT")
5
- os.chdir('GroupViT')
6
 
7
  import os.path as osp
8
  from collections import namedtuple
@@ -23,6 +23,7 @@ from segmentation.evaluation import (GROUP_PALETTE, build_seg_demo_pipeline,
23
  build_seg_inference)
24
  from utils import get_config, load_checkpoint
25
 
 
26
  checkpoint_url = 'https://github.com/xvjiarui/GroupViT-1/releases/download/v1.0.0/group_vit_gcc_yfcc_30e-74d335e6.pth'
27
  cfg_path = 'configs/group_vit_gcc_yfcc_30e.yml'
28
  output_dir = 'demo/output'
 
2
  import os
3
  import sys
4
  os.system("git clone https://github.com/NVlabs/GroupViT")
5
+ sys.path.insert(0, 'GroupViT')
6
 
7
  import os.path as osp
8
  from collections import namedtuple
 
23
  build_seg_inference)
24
  from utils import get_config, load_checkpoint
25
 
26
+ os.chdir('GroupViT')
27
  checkpoint_url = 'https://github.com/xvjiarui/GroupViT-1/releases/download/v1.0.0/group_vit_gcc_yfcc_30e-74d335e6.pth'
28
  cfg_path = 'configs/group_vit_gcc_yfcc_30e.yml'
29
  output_dir = 'demo/output'