jhj0517 commited on
Commit
bb7b1b2
1 Parent(s): dcca9e0

set max duration

Browse files
Files changed (1) hide show
  1. pose_align.py +1 -1
pose_align.py CHANGED
@@ -39,7 +39,7 @@ class PoseAlignmentInference:
39
  if not os.path.exists(self.output_dir):
40
  os.makedirs(self.output_dir)
41
 
42
- @spaces.GPU
43
  def align_pose(
44
  self,
45
  vidfn: str,
 
39
  if not os.path.exists(self.output_dir):
40
  os.makedirs(self.output_dir)
41
 
42
+ @spaces.GPU(duration=120)
43
  def align_pose(
44
  self,
45
  vidfn: str,