Spaces:
Running
Running
Fix ZeroGPU Bug
Browse files
app.py
CHANGED
@@ -238,7 +238,7 @@ class PoseEstimationApp:
|
|
238 |
out.release()
|
239 |
return output_path, new_fps, frames
|
240 |
|
241 |
-
@spaces.GPU
|
242 |
def run_pose_estimation(self, tmp_dir, video_path: str) -> str:
|
243 |
base_name = os.path.splitext(os.path.basename(video_path))[0]
|
244 |
out_dir = os.path.join(tmp_dir, "output")
|
|
|
238 |
out.release()
|
239 |
return output_path, new_fps, frames
|
240 |
|
241 |
+
@spaces.GPU()
|
242 |
def run_pose_estimation(self, tmp_dir, video_path: str) -> str:
|
243 |
base_name = os.path.splitext(os.path.basename(video_path))[0]
|
244 |
out_dir = os.path.join(tmp_dir, "output")
|