visualization

main.py

@@ -1,25 +1,27 @@
 import mmpose
 import os
 from mmpose.apis import MMPoseInferencer
-
 print("[INFO]: Imported modules!")
 
 import gradio as gr
 import numpy as np
 
-def poses(photo):
-    inferencer = MMPoseInferencer('human')
+inferencer = MMPoseInferencer('human')
+print("[INFO]: Downloaded models!")
+
 
-    print("[INFO]: Downloaded models!")
+def poses(photo):
     print(photo)
     result_generator = inferencer(photo, 
-                                  out_dir =".",
+                                  vis_out_dir =".",
                                   return_vis=True)
-
     print("[INFO]: Visualizing results!")
+
     # The MMPoseInferencer API employs a lazy inference approach,
     # creating a prediction generator when given input
     result = next(result_generator)
+    print(os.listdir())
+    print(type(result["visualization"]))
     return result["visualization"]
 
 # # specify detection model by alias