Update app.py
Browse files
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(
|