qihfang commited on
Commit
8faa71f
·
1 Parent(s): 3a251d5

Fix ZeroGPU Bug

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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")