Spaces:
Runtime error
Runtime error
Jyothish CHANDRASENAN
commited on
Commit
·
3d8155a
1
Parent(s):
6280a7e
Update app.py
Browse files
app.py
CHANGED
|
@@ -152,7 +152,7 @@ def predict(inp):
|
|
| 152 |
# infer
|
| 153 |
prediction = infer(preprocessed_img)
|
| 154 |
keypoints_map = postprocessor(prediction, orig_img)
|
| 155 |
-
cv2.imwrite("keypoints_map.png"
|
| 156 |
|
| 157 |
return confidences
|
| 158 |
|
|
|
|
| 152 |
# infer
|
| 153 |
prediction = infer(preprocessed_img)
|
| 154 |
keypoints_map = postprocessor(prediction, orig_img)
|
| 155 |
+
cv2.imwrite("keypoints_map.png", keypoints_map[:,:,::-1])
|
| 156 |
|
| 157 |
return confidences
|
| 158 |
|