Merge branch 'main' of https://huggingface.co/spaces/caizhongang/SMPLer-X

README.md

@@ -4,7 +4,7 @@ emoji: ⚡
 colorFrom: blue
 colorTo: indigo
 sdk: gradio
-python_version: 3.8
+python_version: 3.9
 sdk_version: 4.16.0
 app_file: app.py
 pinned: false

app.py

@@ -27,7 +27,7 @@ print(torch.cuda.get_device_name(index))
 from main.inference import Inferer
 inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
 
-@spaces.GPU
+@spaces.GPU(enable_queue=True)
 def infer(video_input, in_threshold=0.5, num_people="Single person", render_mesh=False):
     os.system(f'rm -rf {OUT_FOLDER}/*')
     multi_person = False if (num_people == "Single person") else True
@@ -131,4 +131,4 @@ with gr.Blocks(title="SMPLer-X", css=".gradio-container") as demo:
         inputs=[video_input, 0.5])
 
 #demo.queue()
-demo.launch(debug=True)
+demo.queue().launch(debug=True)