No CUDA GPUs fix attempt

#3
by cbensimon HF Staff - opened
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -120,15 +120,14 @@ import cv2
120
  from PIL import Image
121
 
122
  from dwpose import DwposeDetector
123
- processor = DwposeDetector.from_pretrained_default() #load pose processor by default
124
 
 
125
  _depth_processor = None
126
 
127
 
128
  def _get_pose_processor():
129
  global _pose_processor
130
  if _pose_processor is None:
131
- from dwpose import DwposeDetector
132
  _pose_processor = DwposeDetector.from_pretrained_default()
133
  print("[Preprocess] DWPose processor loaded")
134
  return _pose_processor
@@ -173,7 +172,7 @@ def extract_first_frame(video_path: str) -> str:
173
 
174
  def preprocess_video_pose(frames: list[np.ndarray], width: int, height: int) -> list[np.ndarray]:
175
  """Extract DWPose skeletons from each frame. Returns float [0,1] frames."""
176
-
177
  result = []
178
  for frame in frames:
179
  pil = Image.fromarray(frame.astype(np.uint8)).convert("RGB")
 
120
  from PIL import Image
121
 
122
  from dwpose import DwposeDetector
 
123
 
124
+ _pose_processor = None
125
  _depth_processor = None
126
 
127
 
128
  def _get_pose_processor():
129
  global _pose_processor
130
  if _pose_processor is None:
 
131
  _pose_processor = DwposeDetector.from_pretrained_default()
132
  print("[Preprocess] DWPose processor loaded")
133
  return _pose_processor
 
172
 
173
  def preprocess_video_pose(frames: list[np.ndarray], width: int, height: int) -> list[np.ndarray]:
174
  """Extract DWPose skeletons from each frame. Returns float [0,1] frames."""
175
+ processor = _get_pose_processor()
176
  result = []
177
  for frame in frames:
178
  pil = Image.fromarray(frame.astype(np.uint8)).convert("RGB")