3d visualization

main.py

@@ -28,7 +28,7 @@ print("[INFO]: Downloaded models!")
 
 
 def poses(photo):
-    print(photo)
+    print("[INFO]: Running inference!")
     result_generator = inferencer(photo, 
                                   vis_out_dir =".",
                                   return_vis=True,
@@ -61,6 +61,8 @@ def poses(photo):
 
     return output_file
 
+
+
 # # specify detection model by alias
 # # the available aliases include 'human', 'hand', 'face', 'animal',
 # # as well as any additional aliases defined in mmdet
@@ -70,12 +72,32 @@ def poses(photo):
 #     pose2d_weights='custom_human_pose_estimator.pth',
 #     det_model='human'
 # )
+
     
 def run():
     #https://github.com/open-mmlab/mmpose/blob/main/docs/en/user_guides/inference.md
-    demo = gr.Interface(fn=poses, 
-                        inputs=gr.Video(source="upload"),
-                        outputs=gr.Video())
+
+    webcam = gr.Interface(
+    fn=poses,
+    inputs= gr.Video(source="webcam"),
+    outputs =  gr.PlayableVideo(format='mp4', interactive=True),
+    title = 'Pose estimation', 
+    description = 'Pose estimation on video',
+    allow_flagging=False
+    )
+
+    file = gr.Interface(
+        poses,
+        inputs = gr.Video(source="upload"),
+        outputs = gr.PlayableVideo(format='mp4', interactive=True),
+        #examples=os.path.join(os.path.dirname(__file__), "examples/aseptic_example.mp4"),
+        allow_flagging=False
+    )
+    demo = gr.TabbedInterface(
+            interface_list=[file, webcam],
+            tab_names=["From a File", "From your Webcam"]
+        )
+
 
 
     demo.launch(server_name="0.0.0.0", server_port=7860)