file display

main.py

@@ -1,17 +1,26 @@
+# Pose inferencing
 import mmpose
-import os
-import glob
 from mmpose.apis import MMPoseInferencer
+
+# Gradio
 import gradio as gr
+
+# System and files
+import os
+import glob
+import uuid
+
+# Image manipulation
 import numpy as np
 import cv2
-import uuid
 
 print("[INFO]: Imported modules!")
 human = MMPoseInferencer("human")
 hand = MMPoseInferencer("hand")
 human3d = MMPoseInferencer(pose3d="human3d")
+# ultraltics
 
+# Defining inferencer models to lookup in function
 inferencers = {"Estimate human 2d poses":human, "Estimate human 2d hand poses":hand, "Estimate human 3d poses":human3d}
 
 # inferencer = MMPoseInferencer('human')
@@ -75,7 +84,7 @@ def run():
     webcam = gr.Interface(
         fn=poses,
         inputs= [gr.Video(source="webcam", height=412), check_web],
-        outputs = [gr.PlayableVideo()]*len(check_web),  #file_types=['.mp4'] #gr.Video(),
+        outputs = [gr.PlayableVideo()]*len(available_methods),  #file_types=['.mp4'] #gr.Video(),
         title = 'Pose estimation', 
         description = 'Pose estimation on video',
         allow_flagging=False
@@ -84,7 +93,7 @@ def run():
     file = gr.Interface(
         poses,
         inputs = [gr.Video(source="upload", height=412), check_file],
-        outputs = [gr.PlayableVideo()]*len(check_file),
+        outputs = [gr.PlayableVideo()]*len(available_methods),
         allow_flagging=False
     )