PengWeixuanSZU commited on
Commit
25439fd
·
verified ·
1 Parent(s): 936ced6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -178,7 +178,7 @@ def preprocess_for_removal(images, masks):
178
  arr_masks = np.stack(out_masks)
179
  return torch.from_numpy(arr_images).half(), torch.from_numpy(arr_masks).half()
180
 
181
- @spaces.GPU(duration=50)
182
  def inference_and_return_video(dilation_iterations, num_inference_steps, video_state):
183
  if video_state["origin_images"] is None or video_state["masks"] is None:
184
  return None
 
178
  arr_masks = np.stack(out_masks)
179
  return torch.from_numpy(arr_images).half(), torch.from_numpy(arr_masks).half()
180
 
181
+ @spaces.GPU(duration=60)
182
  def inference_and_return_video(dilation_iterations, num_inference_steps, video_state):
183
  if video_state["origin_images"] is None or video_state["masks"] is None:
184
  return None