Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -230,8 +230,9 @@ def run(
|
|
230 |
mp_pose.POSE_CONNECTIONS,
|
231 |
landmark_drawing_spec=mp_drawing_styles.get_default_pose_landmarks_style(),
|
232 |
)
|
233 |
-
|
234 |
-
|
|
|
235 |
|
236 |
|
237 |
model_complexities = list(range(3))
|
@@ -250,7 +251,7 @@ demo = gr.Interface(
|
|
250 |
gr.Radio(label="Background Color", choices=background_colors, type="value", value=background_colors[0]),
|
251 |
],
|
252 |
|
253 |
-
outputs=[gr.Image(label="Output"),
|
254 |
examples=examples,
|
255 |
title=TITLE,
|
256 |
description=DESCRIPTION,
|
|
|
230 |
mp_pose.POSE_CONNECTIONS,
|
231 |
landmark_drawing_spec=mp_drawing_styles.get_default_pose_landmarks_style(),
|
232 |
)
|
233 |
+
res, pose_classification = classify_pose(results.pose_landmarks.landmark, res) #Pose Classification code
|
234 |
+
|
235 |
+
return res[:, :, ::-1]
|
236 |
|
237 |
|
238 |
model_complexities = list(range(3))
|
|
|
251 |
gr.Radio(label="Background Color", choices=background_colors, type="value", value=background_colors[0]),
|
252 |
],
|
253 |
|
254 |
+
outputs=[gr.Image(label="Output"),
|
255 |
examples=examples,
|
256 |
title=TITLE,
|
257 |
description=DESCRIPTION,
|