Spaces:
Runtime error
Runtime error
fixed path
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@
|
|
2 |
import os
|
3 |
import sys
|
4 |
os.system("git clone https://github.com/NVlabs/GroupViT")
|
5 |
-
|
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'
|