gr state on human3d

main_noweb.py

@@ -32,6 +32,10 @@ print("[INFO]: Imported modules!")
 human = MMPoseInferencer("simcc_mobilenetv2_wo-deconv-8xb64-210e_coco-256x192") # simcc_mobilenetv2_wo-deconv-8xb64-210e_coco-256x192 dekr_hrnet-w32_8xb10-140e_coco-512x512
 hand = MMPoseInferencer("hand")
 #model3d = gr.State()
+human3d = MMPoseInferencer(device=device, 
+                               pose3d="human3d", 
+                               scope="mmpose")
+
 
 #"https://github.com/open-mmlab/mmpose/blob/main/configs/body_3d_keypoint/pose_lift/h36m/pose-lift_simplebaseline3d_8xb64-200e_h36m.py", 
                            #"https://download.openmmlab.com/mmpose/body3d/simple_baseline/simple3Dbaseline_h36m-f0ad73a4_20210419.pth") # pose3d="human3d"
@@ -90,15 +94,13 @@ def check_extension(video):
     return video
 
 
-def pose3d(video, kpt_threshold):
+def pose3d(video, kpt_threshold, ):
+
     video = check_extension(video)
     print(device)
     
     #human3d = MMPoseInferencer(device=device,   pose3d="human3d",  scope="mmpose")#"pose-lift_videopose3d-243frm-supv-cpn-ft_8xb128-200e_h36m")
 
-    human3d = MMPoseInferencer(device=device, 
-                               pose3d="human3d", 
-                               scope="mmpose")
     print("HUMAN 3d downloaded!!")
     #gr.State(value=)
     
@@ -109,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 = human3d(video, 
+    result_generator = gr.State(human3d(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)        
 
@@ -195,6 +197,7 @@ def pose2dhand(video, kpt_threshold):
 
 def UI():
     block = gr.Blocks()
+
     with block:
         with gr.Column():       
             with gr.Tab("Upload video"):