visualization

main.py

@@ -1,6 +1,7 @@
 import mmpose
-print(mmpose.__version__)
+import os
 from mmpose.apis import MMPoseInferencer
+
 inferencer = MMPoseInferencer('human')
 print("[INFO]: Imported modules!")
 
@@ -11,9 +12,10 @@ def greet(photo):
     result_generator = inferencer(photo)
 
     print("[INFO]: Visualizing results!")
-    vis, pred = next(result_generator)
-    print(vis)
-    return vis
+    result = next(result_generator)
+    print(os.listdir())
+    print(result['visualization'])
+    return result['visualization']
 
 # # specify detection model by alias
 # # the available aliases include 'human', 'hand', 'face', 'animal',