Piarasingh85 commited on
Commit
49a721f
·
verified ·
1 Parent(s): 751ce7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -125,6 +125,7 @@ def detect_and_classify_pose(input_image, complexity, confidence, background_col
125
  mp_drawing.draw_landmarks(input_image, results.pose_landmarks, mp_pose.POSE_CONNECTIONS)
126
  input_image, pose_classification = classify_pose(results.pose_landmarks.landmark, input_image)
127
 
 
128
  return input_image, pose_classification
129
 
130
  iface = gr.Interface(
 
125
  mp_drawing.draw_landmarks(input_image, results.pose_landmarks, mp_pose.POSE_CONNECTIONS)
126
  input_image, pose_classification = classify_pose(results.pose_landmarks.landmark, input_image)
127
 
128
+ input_image = cv2.cvtColor(input_image, cv2.COLOR_RGB2BGR) # Convert back to BGR for correct display in Gradio
129
  return input_image, pose_classification
130
 
131
  iface = gr.Interface(