visualization

Dockerfile

@@ -24,6 +24,7 @@ WORKDIR /..
 # RUN mim install "mmpose>=1.1.0"
 
 RUN pip install --no-cache-dir --upgrade gradio
+RUN pip install --no-cache-dir --upgrade numpy
 
 #https://stackoverflow.com/questions/55313610/importerror-libgl-so-1-cannot-open-shared-object-file-no-such-file-or-directo
 RUN apt-get update && apt-get install ffmpeg libsm6 libxext6  -y

main.py

@@ -5,6 +5,7 @@ from mmpose.apis import MMPoseInferencer
 print("[INFO]: Imported modules!")
 
 import gradio as gr
+import numpy as np
 
 def poses(photo):
     inferencer = MMPoseInferencer('human')
@@ -18,12 +19,8 @@ def poses(photo):
     # The MMPoseInferencer API employs a lazy inference approach,
     # creating a prediction generator when given input
     result = next(result_generator)
-    print(type(result))
-    print(result['visualization'])
-    print(type(result['visualization']))
-    print(result['visualization'][0])
-    print(type(result['visualization'][0]))
-    return result['visualization'][0]
+    output = np.array(result["visualization"])
+    return output
 
 # # specify detection model by alias
 # # the available aliases include 'human', 'hand', 'face', 'animal',