PengWeixuanSZU commited on
Commit
f3d8e03
·
verified ·
1 Parent(s): 0d4e4ae

Update app.py

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