Spaces:

basementparking
/

car

Sleeping

basementparking commited on Jul 22, 2024

Commit

e813ad0

verified ·

1 Parent(s): 1e1fdd6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ from detectron2.data import MetadataCatalog, DatasetCatalog
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-dist = distutils.core.run_setup("./detectron2/setup.py")
 def setup_model():
     cfg = get_cfg()
     cfg.MODEL.DEVICE = "cuda" if torch.cuda.is_available() else "cpu"

 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+dist = distutils.core.run_setup("./detectron2-main/setup.py")
 def setup_model():
     cfg = get_cfg()
     cfg.MODEL.DEVICE = "cuda" if torch.cuda.is_available() else "cpu"