dagloop5 commited on
Commit
5622309
·
verified ·
1 Parent(s): 9342ac4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -609,7 +609,7 @@ def on_highres_toggle(first_image, last_image, high_res):
609
  def get_gpu_duration(gpu_duration: float):
610
  return int(gpu_duration)
611
 
612
- @spaces.GPU(duration=gpu_duration)
613
  @torch.inference_mode()
614
  def generate_video(
615
  first_image,
 
609
  def get_gpu_duration(gpu_duration: float):
610
  return int(gpu_duration)
611
 
612
+ @spaces.GPU(duration=get_gpu_duration)
613
  @torch.inference_mode()
614
  def generate_video(
615
  first_image,