Spaces:
Paused
Paused
updated code from cuda to cpu
Browse files
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")
|