Update scripts/demo.py

scripts/demo.py

@@ -1,6 +1,6 @@
 import os
 import sys
-os.environ['PYOPENGL_PLATFORM'] = 'egl'
+os.environ['PYOPENGL_PLATFORM'] = 'osmesa'
 os.environ['CUDA_VISIBLE_DEVICES'] = 'cpu'
 sys.path.append(os.getcwd())
 
@@ -250,7 +250,7 @@ def infer(g_body, g_face, smplx_model, rendertool, config, args):
 def main():
     parser = parse_args()
     args = parser.parse_args()
-    device = torch.device(args.gpu)
+    device = torch.device(args.cpu)
     torch.cuda.set_device(device)
 
     config = load_JsonConfig(args.config_file)