Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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.
|
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 \
|