Spaces:
Build error
Build error
nishantkaushik20
commited on
Commit
·
ff8f460
1
Parent(s):
a6fa343
Update src/classification_keypoint.py
Browse files
src/classification_keypoint.py
CHANGED
@@ -43,7 +43,7 @@ class KeypointClassification:
|
|
43 |
|
44 |
if __name__ == '__main__':
|
45 |
keypoint_classification = KeypointClassification(
|
46 |
-
path_model='/Users/
|
47 |
)
|
48 |
dummy_input = torch.randn(23)
|
49 |
classification = keypoint_classification(dummy_input)
|
|
|
43 |
|
44 |
if __name__ == '__main__':
|
45 |
keypoint_classification = KeypointClassification(
|
46 |
+
path_model='/Users/nishantkaushik20/Me/source-code/AI/PoseEstimationYOLOv8/models/pose_classification.pt'
|
47 |
)
|
48 |
dummy_input = torch.randn(23)
|
49 |
classification = keypoint_classification(dummy_input)
|