gr state on human3d

main_noweb.py

@@ -94,7 +94,7 @@ def check_extension(video):
     return video
 
 
-def pose3d(video, kpt_threshold, model):
+def pose3d(video, kpt_threshold):
 
     video = check_extension(video)
     print(device)
@@ -109,7 +109,7 @@ def pose3d(video, kpt_threshold, model):
     os.makedirs(add_dir)
     print(check_fps(video))
     #video = human3d.preprocess(video, batch_size=8)
-    result_generator = gr.State(model(video, 
+    result_generator = gr.State(human3d(video, 
                             vis_out_dir = add_dir,
                             radius = 8,
                             thickness = 5,
@@ -156,11 +156,11 @@ def pose2d(video, kpt_threshold):
 
     return "".join(out_file), "".join(kpoints)
 
-def pose3dbatch(video, kpt_threshold, model):
+def pose3dbatch(video, kpt_threshold):
     kpoints=[]
     outvids=[]
-    for v, t in zip(video, kpt_threshold, model):
-        vname, kname = pose3d(v, t, model)
+    for v, t in zip(video, kpt_threshold):
+        vname, kname = pose3d(v, t)
         outvids.append(vname) 
         kpoints.append(kname)
     return [outvids]#kpoints, outvids
@@ -321,7 +321,7 @@ print("Angle is: ", angle)
                             queue=True)
         
         submit_pose3d_file.click(fn=pose3dbatch, 
-                                inputs= [video_input, file_kpthr, human3d], 
+                                inputs= [video_input, file_kpthr], 
                                 outputs = video_output2,#[video_output2, jsonoutput],
                                 batch=True, 
                                 max_batch_size=16,