Hugging Face's logo

Spaces:
purkrmir
/
ProbPose-demo
Sleeping

App Files Community
Miroslav Purkrabek commited on
Commit
8f751fc
·
1 Parent(s): f41366b

always use GPU

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ DET_WEIGHTS = "https://download.openmmlab.com/mmpose/v1/projects/rtmpose/rtmdet_
42
  POSE_CFG = "configs/body_2d_keypoint/topdown_probmap/coco/td-pm_ProbPose-small_8xb64-210e_coco-256x192.py"
43
  POSE_WEIGHTS = "models/ProbPose-s.pth"
44
 
45
- DEVICE = 'cuda:0' if spaces.is_gpu_available() else 'cpu'
46
 
47
  @spaces.GPU
48
  def init_models():
 
42
  POSE_CFG = "configs/body_2d_keypoint/topdown_probmap/coco/td-pm_ProbPose-small_8xb64-210e_coco-256x192.py"
43
  POSE_WEIGHTS = "models/ProbPose-s.pth"
44
 
45
+ DEVICE = 'cuda:0'
46
 
47
  @spaces.GPU
48
  def init_models():