Fabrice-TIERCELIN commited on
Commit
cafbe8a
1 Parent(s): bf3b935

"480 is more than the time allowed": we have to reduce the inference steps

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,7 +19,7 @@ interface = gr.Interface(
19
  theme="soft"
20
  )
21
 
22
- @spaces.GPU(duration=480)
23
  def generate_video(image):
24
  """
25
  Generates a video from an input image using the pipeline.
@@ -30,7 +30,7 @@ def generate_video(image):
30
  Returns:
31
  The path of a video file.
32
  """
33
- video_frames = pipeline(image=image, num_inference_steps=20).images
34
 
35
  # Frames to Video
36
  os.makedirs("outputs", exist_ok=True)
 
19
  theme="soft"
20
  )
21
 
22
+ @spaces.GPU(duration=360)
23
  def generate_video(image):
24
  """
25
  Generates a video from an input image using the pipeline.
 
30
  Returns:
31
  The path of a video file.
32
  """
33
+ video_frames = pipeline(image=image, num_inference_steps=10).images
34
 
35
  # Frames to Video
36
  os.makedirs("outputs", exist_ok=True)