LPX55 commited on
Commit
51d25bc
·
verified ·
1 Parent(s): b6c95c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -100,7 +100,7 @@ def resize_image_to_bucket(image: Union[Image.Image, np.ndarray], bucket_reso: T
100
  return image
101
 
102
 
103
- @spaces.GPU()
104
  def generate_video(prompt: str, frame1: Image.Image, frame2: Image.Image, resolution: str, guidance_scale: float, num_frames: int, num_inference_steps: int) -> bytes:
105
  # Debugging print statements
106
  print(f"Frame 1 Type: {type(frame1)}")
 
100
  return image
101
 
102
 
103
+ @spaces.GPU(duration=120)
104
  def generate_video(prompt: str, frame1: Image.Image, frame2: Image.Image, resolution: str, guidance_scale: float, num_frames: int, num_inference_steps: int) -> bytes:
105
  # Debugging print statements
106
  print(f"Frame 1 Type: {type(frame1)}")