BSuruchi commited on
Commit
f788148
·
verified ·
1 Parent(s): eedb724

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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
- # res, pose_classification = classify_pose(results.pose_landmarks.landmark, res) #Pose Classification code
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