ttxskk commited on
Commit
c434dc1
·
verified ·
1 Parent(s): 32fe019

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -48,7 +48,7 @@ os.makedirs(OUT_FOLDER, exist_ok=True)
48
  DEMO_CONFIG = '/home/user/app/config/aios_smplx_demo.py'
49
  MODEL_PATH = '/home/user/app/pretrained_models/aios_checkpoint.pth'
50
  @spaces.GPU(enable_queue=True, duration=300)
51
- def infer(video_input, batch_size, threshold=0.5, num_person=1):
52
  os.system(f'rm -rf {OUT_FOLDER}/*')
53
  os.system(f'torchrun --nproc_per_node 1 \
54
  main.py \
 
48
  DEMO_CONFIG = '/home/user/app/config/aios_smplx_demo.py'
49
  MODEL_PATH = '/home/user/app/pretrained_models/aios_checkpoint.pth'
50
  @spaces.GPU(enable_queue=True, duration=300)
51
+ def infer(video_input, batch_size, threshold=0.3, num_person=1):
52
  os.system(f'rm -rf {OUT_FOLDER}/*')
53
  os.system(f'torchrun --nproc_per_node 1 \
54
  main.py \