hpoghos commited on
Commit
2536d14
1 Parent(s): fd47cf9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -43,7 +43,7 @@ examples = [
43
  ["",
44
  "200 - frames (recommended)", 33, "__assets__/gradio_cached_examples/test11.jpg", None],
45
  ]
46
- @spaces.GPU
47
  def generate(prompt, num_frames, seed, image: np.ndarray):
48
  if num_frames == [] or num_frames is None:
49
  num_frames = 50
@@ -66,7 +66,7 @@ def generate(prompt, num_frames, seed, image: np.ndarray):
66
 
67
  return image, video_file_stage_one, video_file_stage_two
68
 
69
- @spaces.GPU
70
  def enhance(prompt, num_frames, seed, image: np.ndarray, video:str):
71
  if num_frames == [] or num_frames is None:
72
  num_frames = 50
 
43
  ["",
44
  "200 - frames (recommended)", 33, "__assets__/gradio_cached_examples/test11.jpg", None],
45
  ]
46
+ @spaces.GPU(duration=2000)
47
  def generate(prompt, num_frames, seed, image: np.ndarray):
48
  if num_frames == [] or num_frames is None:
49
  num_frames = 50
 
66
 
67
  return image, video_file_stage_one, video_file_stage_two
68
 
69
+ @spaces.GPU(duration=2000)
70
  def enhance(prompt, num_frames, seed, image: np.ndarray, video:str):
71
  if num_frames == [] or num_frames is None:
72
  num_frames = 50