gr state on human3d

main_noweb.py

@@ -32,9 +32,9 @@ 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 = gr.State(MMPoseInferencer(device=device, 
+human3d = MMPoseInferencer(device=device, 
                                pose3d="human3d", 
-                               scope="mmpose"))
+                               scope="mmpose")
 
 
 #"https://github.com/open-mmlab/mmpose/blob/main/configs/body_3d_keypoint/pose_lift/h36m/pose-lift_simplebaseline3d_8xb64-200e_h36m.py", 
@@ -109,14 +109,14 @@ def pose3d(video, kpt_threshold, model):
     os.makedirs(add_dir)
     print(check_fps(video))
     #video = human3d.preprocess(video, batch_size=8)
-    result_generator = model(video, 
+    result_generator = gr.State(model(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)