Update app.py
Browse files
app.py
CHANGED
@@ -231,8 +231,10 @@ def run(
|
|
231 |
mp_pose.POSE_CONNECTIONS,
|
232 |
landmark_drawing_spec=mp_drawing_styles.get_default_pose_landmarks_style(),
|
233 |
)
|
234 |
-
|
235 |
-
|
|
|
|
|
236 |
return res[:, :, ::-1]
|
237 |
|
238 |
|
|
|
231 |
mp_pose.POSE_CONNECTIONS,
|
232 |
landmark_drawing_spec=mp_drawing_styles.get_default_pose_landmarks_style(),
|
233 |
)
|
234 |
+
|
235 |
+
if results.pose_landmarks:
|
236 |
+
res, pose_classification = classify_pose(results.pose_landmarks.landmark, res) #Pose Classification code
|
237 |
+
|
238 |
return res[:, :, ::-1]
|
239 |
|
240 |
|