update: gpu supports

lib/kits/gradio/hsmr_service.py

@@ -59,8 +59,8 @@ class HSMRService:
             # 1a. Setup UI.
             gr.Markdown(self.description)
 
-            with gr.Tab(label='HSMR-IMG-CPU'):
-                gr.Markdown('> **Pure CPU** demo for recoverying human mesh and skeleton from a single image. Each inference may take **about 3 minutes**.')
+            with gr.Tab(label='HSMR-IMG'):
+                gr.Markdown('> Demo for recoverying human mesh and skeleton from a single image. (For **Pure CPU** demo, inference may take **about 1~3 minutes**.)')
                 with gr.Row(equal_height=False):
                     with gr.Column():
                         input_image = gr.Image(

tools/service.py

@@ -1,12 +1,14 @@
 from lib.kits.gradio import *
 
 import os
+import torch
 
 os.environ['PYOPENGL_PLATFORM'] = 'egl'
 
 
 if __name__ == '__main__':
     # Start serving.
-    backend = HSMRBackend(device='cpu')
+    device = 'cuda' if torch.cuda.is_available() else 'cpu'
+    backend = HSMRBackend(device=device)
     hsmr_service = HSMRService(backend)
     hsmr_service.serve()