file display

main.py

@@ -32,28 +32,28 @@ print("[INFO]: Downloaded models!")
 
 def poses(photo, check):
     # Selecting the specific inferencer
+    out_files=[]
+    for i in len(check):
+        inferencer = inferencers[check] # 'hand', 'human , device='cuda'
 
-    inferencer = inferencers[check[0]] # 'hand', 'human , device='cuda'
+        print("[INFO]: Running inference!")
+        # Create out directory
+        vis_out_dir = str(uuid.uuid4())
 
-    print("[INFO]: Running inference!")
-    # Create out directory
-    vis_out_dir = str(uuid.uuid4())
-
-    result_generator = inferencer(photo, 
-                                  vis_out_dir = vis_out_dir,
-                                  return_vis=True,
-                                  thickness=2,
-                                  rebase_keypoint_height=True)    
-    
-    result = [result for result in result_generator] #next(result_generator)    
-
-    output_files = glob.glob(os.path.join(vis_out_dir, "*.mp4"))
-    # 00000.mp4
-    # 000000.mp4
-
-    return output_files[0]
+        result_generator = inferencer(photo, 
+                                    vis_out_dir = vis_out_dir,
+                                    return_vis=True,
+                                    thickness=2,
+                                    rebase_keypoint_height=True)    
+        
+        result = [result for result in result_generator] #next(result_generator)    
 
+        out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4"))
+        # 00000.mp4
+        # 000000.mp4
+        out_files.append(out_file)
 
+    return out_files
 
 # # specify detection model by alias
 # # the available aliases include 'human', 'hand', 'face', 'animal',
@@ -75,7 +75,7 @@ def run():
     webcam = gr.Interface(
         fn=poses,
         inputs= [gr.Video(source="webcam", height=412), check_web],
-        outputs = gr.PlayableVideo(),  #file_types=['.mp4'] #gr.Video(),
+        outputs = [gr.PlayableVideo()]*len(check_web),  #file_types=['.mp4'] #gr.Video(),
         title = 'Pose estimation', 
         description = 'Pose estimation on video',
         allow_flagging=False
@@ -84,7 +84,7 @@ def run():
     file = gr.Interface(
         poses,
         inputs = [gr.Video(source="upload", height=412), check_file],
-        outputs = gr.PlayableVideo(),
+        outputs = [gr.PlayableVideo()]*len(check_file),
         allow_flagging=False
     )