Spaces:
Running
on
Zero
Running
on
Zero
Add @spaces.GPU to detect_pose_video
Browse files
app.py
CHANGED
@@ -173,6 +173,10 @@ def detect_pose_image(
|
|
173 |
return annotated_frame, human_readable_results
|
174 |
|
175 |
|
|
|
|
|
|
|
|
|
176 |
def detect_pose_video(
|
177 |
video_path: str,
|
178 |
threshold: float,
|
|
|
173 |
return annotated_frame, human_readable_results
|
174 |
|
175 |
|
176 |
+
# Decorate this function with `@spaces.GPU` to ensure that ZeroGPU is allocated once for the entire video processing.
|
177 |
+
# Although `detect_pose_image` (called per frame) is already decorated, without this decorator, ZeroGPU would be invoked for each frame,
|
178 |
+
# causing significant overhead and slowdowns. By decorating this function, all frames are processed sequentially after a single GPU allocation.
|
179 |
+
@spaces.GPU(duration=90)
|
180 |
def detect_pose_video(
|
181 |
video_path: str,
|
182 |
threshold: float,
|