markdown slider and max threads

main_noweb.py

@@ -120,7 +120,7 @@ def pose3d(video, kpt_threshold):
     print(kpoints) 
 
     # Reinitialize
-    return "".join(out_file)
+    return "".join(out_file),  "".join(kpoints)
 
 
 def pose2d(video, kpt_threshold):
@@ -129,7 +129,6 @@ def pose2d(video, kpt_threshold):
 
     # Define new unique folder
     add_dir = str(uuid.uuid4())
-    vis_out_dir = os.path.join("/".join(video.split("/")[:-1]), add_dir)
 
     os.makedirs(add_dir)
     print(check_fps(video))
@@ -160,23 +159,25 @@ def pose2dhand(video, kpt_threshold):
 
     # Define new unique folder
     add_dir = str(uuid.uuid4())
-    vis_out_dir = os.path.join("/".join(video.split("/")[:-1]), add_dir)
-    os.makedirs(vis_out_dir)
+    os.makedirs(add_dir)
 
     result_generator = hand(video, 
-                                 vis_out_dir = vis_out_dir,
-                                 return_vis=True,
+                                 vis_out_dir = add_dir,
                                  thickness = 4,
                                  radius = 5,
                                  rebase_keypoint_height=True,
                                  kpt_thr=kpt_threshold,
+                                 pred_out_dir = add_dir, 
+
                                  device=device)    
     
     result = [result for result in result_generator] #next(result_generator)        
 
-    out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm")) 
+    out_file = glob.glob(os.path.join(add_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm")) 
+    kpoints = glob.glob(os.path.join(add_dir, "*.json"))
+
    
-    return "".join(out_file)
+    return "".join(out_file), "".join(kpoints)
 
 def UI():
     block = gr.Blocks()