mrcuddle commited on
Commit
1977741
·
verified ·
1 Parent(s): 268facb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ pipeline = StableVideoDiffusionPipeline.from_pretrained(
13
  )
14
  pipeline.to(device)
15
 
16
- @spaces.GPU
17
  def generate_video(image_path, seed):
18
  # Load and preprocess the image
19
  image = load_image(image_path)
 
13
  )
14
  pipeline.to(device)
15
 
16
+ @spaces.GPU(duration=120)
17
  def generate_video(image_path, seed):
18
  # Load and preprocess the image
19
  image = load_image(image_path)