3d visualization

main.py

@@ -31,9 +31,9 @@ def poses(photo):
                                   vis_out_dir =".",
                                   return_vis=True,
                                   thickness=2)    
-    
-    for result in result_generator:
-         print("[INFO] Result: ", result)
+    result = next(result_generator)
+    #for result in result_generator:
+    #     print("[INFO] Result: ", result)
     # # Prepare to save video
     # output_file = os.path.join("output.mp4")
 
@@ -59,7 +59,7 @@ def poses(photo):
     # cv2.destroyAllWindows() # Closing window
     output_file = glob.glob("*.mp4")
 
-    return output_file
+    return output_file[0]