gr state on human3d

main_noweb.py

@@ -102,7 +102,7 @@ def pose3d(video, kpt_threshold, ):
     #human3d = MMPoseInferencer(device=device,   pose3d="human3d",  scope="mmpose")#"pose-lift_videopose3d-243frm-supv-cpn-ft_8xb128-200e_h36m")
 
     print("HUMAN 3d downloaded!!")
-    #gr.State(value=)
+    human3dst = gr.State(value=human3d)
     
     # Define new unique folder
     add_dir = str(uuid.uuid4())
@@ -111,14 +111,14 @@ def pose3d(video, kpt_threshold, ):
     os.makedirs(add_dir)
     print(check_fps(video))
     #video = human3d.preprocess(video, batch_size=8)
-    result_generator = gr.State(human3d(video, 
+    result_generator = human3dst(video, 
                             vis_out_dir = add_dir,
                             radius = 8,
                             thickness = 5,
                             rebase_keypoint_height=True,
                             kpt_thr=kpt_threshold,
                             pred_out_dir = add_dir
-                            ))    
+                            )
     print("INFERENCE DONW")
     result = [result for result in result_generator] #next(result_generator)