Update preprocess/openpose/annotator/openpose/body.py

preprocess/openpose/annotator/openpose/body.py

@@ -20,8 +20,8 @@ from .model import bodypose_model
 class Body(object):
     def __init__(self, model_path):
         self.model = bodypose_model()
-        if torch.cuda.is_available():
-            self.model = self.model.cuda()
+        # if torch.cuda.is_available():
+        #     self.model = self.model.cuda()
         #     print('cuda')
         model_dict = util.transfer(self.model, torch.load(model_path))
         self.model.load_state_dict(model_dict)