Spaces:

Andy1621
/

uniformer_image_detection

Build error

SakuraD commited on May 23, 2022

Commit

286ae46

•

1 Parent(s): 93f3a7b

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ checkpoint_file = hf_hub_download(repo_id="Andy1621/uniformer", filename="mask_r
 config_file = './exp/mask_rcnn_3x_ms_hybrid_small/config.py'
 # init detector
 # build the model from a config file and a checkpoint file
-model = init_detector(config_file, checkpoint_file, device='cuda:0')
 def set_example_image(example: list) -> dict:

 config_file = './exp/mask_rcnn_3x_ms_hybrid_small/config.py'
 # init detector
 # build the model from a config file and a checkpoint file
+model = init_detector(config_file, checkpoint_file, device='cpu')
 def set_example_image(example: list) -> dict: