Spaces:
Runtime error
Runtime error
Jyothish CHANDRASENAN
commited on
Commit
·
c0d35be
1
Parent(s):
fcd9f6d
Update app.py
Browse files
app.py
CHANGED
|
@@ -152,7 +152,8 @@ def predict(img):
|
|
| 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 |
|
| 158 |
gr.Interface(fn=predict,
|
|
|
|
| 152 |
# infer
|
| 153 |
prediction = infer(preprocessed_img)
|
| 154 |
keypoints_map = postprocessor(prediction, orig_img)
|
| 155 |
+
# cv2.imwrite("keypoints_map.png", keypoints_map[:,:,::-1])
|
| 156 |
+
return keypoints_map
|
| 157 |
|
| 158 |
|
| 159 |
gr.Interface(fn=predict,
|