Hugging Face's logo

Spaces:
basementparking
/
car
Sleeping

App Files Community
basementparking commited on
Commit
bb40947
·
verified ·
1 Parent(s): e813ad0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -25,7 +25,6 @@ dist = distutils.core.run_setup("./detectron2-main/setup.py")
25
  def setup_model():
26
  cfg = get_cfg()
27
  cfg.MODEL.DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
28
- cfg = get_cfg()
29
  cfg.merge_from_file(model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml"))
30
  cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
31
  cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml")
 
25
  def setup_model():
26
  cfg = get_cfg()
27
  cfg.MODEL.DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
28
  cfg.merge_from_file(model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml"))
29
  cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
30
  cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml")