test frontend

app.py

@@ -27,6 +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
+
 @spaces.GPU(enable_queue=True)
 def infer(image_input, in_threshold=0.5, num_people="Single person", render_mesh=False):
     # num_gpus = 1 if torch.cuda.is_available() else -1
@@ -73,7 +75,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
-    return image_input, "success"
+    state_cnt += 1
+    return image_input, "success {}".format(state_cnt)
 
 TITLE = '''<h1 align="center">PostoMETRO: Pose Token Enhanced Mesh Transformer for Robust 3D Human Mesh Recovery</h1>'''
 DESCRIPTION = '''