AnishaNaik03 commited on
Commit
61baaac
·
verified ·
1 Parent(s): 36c849b

updated code from cuda to cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from detectron2.config import get_cfg
9
  from detectron2 import model_zoo
10
  import torch_utils
11
  import dnnlib
12
- Create output directory if it doesn't exist
13
  output_dir = "key/"
14
  os.makedirs(output_dir, exist_ok=True)
15
  output_file = os.path.join(output_dir, "keypoints.json")
 
9
  from detectron2 import model_zoo
10
  import torch_utils
11
  import dnnlib
12
+ # Create output directory if it doesn't exist
13
  output_dir = "key/"
14
  os.makedirs(output_dir, exist_ok=True)
15
  output_file = os.path.join(output_dir, "keypoints.json")