checkbox

main.py

@@ -7,8 +7,12 @@ import numpy as np
 import cv2
 
 print("[INFO]: Imported modules!")
+human = MMPoseInferencer("human")
+hand = MMPoseInferencer("hand")
+human3d = MMPoseInferencer(pose3d="human3d")
+
+inferencers = {"Estimate human 2d poses":human, "Estimate human 2d hand poses":hand, "Estimate human 3d poses":human3d}
 
-inferencer = MMPoseInferencer('hand') # 'hand', 'human , device='cuda'
 # inferencer = MMPoseInferencer('human')
 # inferencer = MMPoseInferencer(pose3d="human3d")
 
@@ -25,7 +29,10 @@ inferencer = MMPoseInferencer('hand') # 'hand', 'human , device='cuda'
 print("[INFO]: Downloaded models!")
 
 
-def poses(photo):
+def poses(photo, check):
+    # Selecting the specific inferencer
+    inferencer = inferencers[check] # 'hand', 'human , device='cuda'
+
     print("[INFO]: Running inference!")
     #vis_out_dir
     result_generator = inferencer(photo, 
@@ -80,9 +87,11 @@ def poses(photo):
     
 def run():
     #https://github.com/open-mmlab/mmpose/blob/main/docs/en/user_guides/inference.md
+    available_methods = ["Estimate human 2d poses", "Estimate human 2d hand poses", "Estimate human 3d poses"]
+    check = gr.CheckboxGroup(available_methods, label="Methods", type="value", info="Select the model(s) you want")
     webcam = gr.Interface(
         fn=poses,
-        inputs= gr.Video(source="webcam"),
+        inputs= [gr.Video(source="webcam"), check],
         outputs =  gr.Video(format='mp4'),
         title = 'Pose estimation', 
         description = 'Pose estimation on video',
@@ -91,10 +100,11 @@ def run():
 
     file = gr.Interface(
         poses,
-        inputs = gr.Video(source="upload"),
+        inputs = [gr.Video(source="upload"), check],
         outputs = gr.Video(format='mp4'),
         allow_flagging=False
     )
+
     demo = gr.TabbedInterface(
             interface_list=[file, webcam],
             tab_names=["From a File", "From your Webcam"]