Hugging Face's logo

codingggasdfasf
/
video-animator

Diffusers
ONNX
Model card Files
xet
Community
1
root commited on
Commit
95a1cba
·
1 Parent(s): bf5ffb9

changing paths

Browse files
Files changed (2) hide show
  1. src/dwpose/__pycache__/wholebody.cpython-310.pyc +0 -0
  2. src/dwpose/wholebody.py +4 -2
src/dwpose/__pycache__/wholebody.cpython-310.pyc CHANGED
Binary files a/src/dwpose/__pycache__/wholebody.cpython-310.pyc and b/src/dwpose/__pycache__/wholebody.cpython-310.pyc differ
 
src/dwpose/wholebody.py CHANGED
@@ -515,8 +515,10 @@ class Wholebody:
515
  )
516
 
517
  base_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
518
- onnx_det = os.path.join(base_dir, "pretrained_weights", "DWPose", "yolox_l.onnx")
519
- onnx_pose = os.path.join(base_dir, "pretrained_weights", "DWPose", "dw-ll_ucoco_384.onnx")
 
 
520
 
521
  self.session_det = ort.InferenceSession(
522
  path_or_bytes=onnx_det, providers=providers
 
515
  )
516
 
517
  base_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
518
+
519
+ # Construct absolute paths for the ONNX files
520
+ onnx_det = os.path.abspath(os.path.join(base_dir, "pretrained_weights", "DWPose", "yolox_l.onnx"))
521
+ onnx_pose = os.path.abspath(os.path.join(base_dir, "pretrained_weights", "DWPose", "dw-ll_ucoco_384.onnx"))
522
 
523
  self.session_det = ort.InferenceSession(
524
  path_or_bytes=onnx_det, providers=providers