Spaces:

Andy1621
/

uniformer_image_segmentation

Running

SakuraD commited on May 23, 2022

Commit

6efffe7

•

1 Parent(s): 8a697b3

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ from huggingface_hub import hf_hub_download
 # Set to cuda to load on GPU
 device = "cpu"
 checkpoint_file = hf_hub_download(repo_id="Andy1621/uniformer", filename="upernet_global_small.pth")
-config_file = './exp/upernet_global_small/test_config_h32.py'
 # init detector
 # build the model from a config file and a checkpoint file
 model = init_segmentor(config_file, checkpoint_file, device='cpu')

 # Set to cuda to load on GPU
 device = "cpu"
 checkpoint_file = hf_hub_download(repo_id="Andy1621/uniformer", filename="upernet_global_small.pth")
+config_file = './exp/upernet_global_small/config.py'
 # init detector
 # build the model from a config file and a checkpoint file
 model = init_segmentor(config_file, checkpoint_file, device='cpu')