test frontend

app.py

@@ -27,7 +27,8 @@ os.makedirs(OUT_FOLDER, exist_ok=True)
 # from main.inference import Inferer
 # inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
 
-state_cnt = 0
+state_cnt = {}
+state_cnt['k'] = 0
 
 @spaces.GPU(enable_queue=True)
 def infer(image_input, in_threshold=0.5, num_people="Single person", render_mesh=False):
@@ -75,8 +76,8 @@ def infer(image_input, in_threshold=0.5, num_people="Single person", render_mesh
     # os.system(f'zip -r {save_mesh_file} {save_path_mesh}')
     # os.system(f'zip -r {save_smplx_file} {save_path_smplx}')
     # yield img, video_path, save_mesh_file, save_smplx_file
-    state_cnt += 1
-    return image_input, "success {}".format(state_cnt)
+    state_cnt['k'] += 1
+    return image_input, "success {}".format(state_cnt['k'])
 
 TITLE = '''<h1 align="center">PostoMETRO: Pose Token Enhanced Mesh Transformer for Robust 3D Human Mesh Recovery</h1>'''
 DESCRIPTION = '''