Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,4 +1,3 @@
|
|
| 1 |
-
import spaces
|
| 2 |
import torch, torchvision
|
| 3 |
import sys
|
| 4 |
# sys.path.insert(0, 'test_mmpose/')
|
|
@@ -28,7 +27,6 @@ pose_model = init_pose_model(pose_config, pose_checkpoint, device='cuda')
|
|
| 28 |
# initialize detector
|
| 29 |
det_model = init_detector(det_config, det_checkpoint, device='cuda')
|
| 30 |
|
| 31 |
-
@spaces.GPU
|
| 32 |
def predict(img):
|
| 33 |
mmdet_results = inference_detector(det_model, img)
|
| 34 |
person_results = process_mmdet_results(mmdet_results, cat_id=1)
|
|
|
|
|
|
|
| 1 |
import torch, torchvision
|
| 2 |
import sys
|
| 3 |
# sys.path.insert(0, 'test_mmpose/')
|
|
|
|
| 27 |
# initialize detector
|
| 28 |
det_model = init_detector(det_config, det_checkpoint, device='cuda')
|
| 29 |
|
|
|
|
| 30 |
def predict(img):
|
| 31 |
mmdet_results = inference_detector(det_model, img)
|
| 32 |
person_results = process_mmdet_results(mmdet_results, cat_id=1)
|