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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -179,7 +179,6 @@ 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=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
@@ -218,7 +217,6 @@ def inference_and_return_video(dilation_iterations, num_inference_steps, video_s
218
  clip.write_videofile(video_file, codec='libx264', audio=False, verbose=False, logger=None)
219
  return video_file
220
 
221
- @spaces.GPU(duration=100)
222
  def track_video(n_frames, video_state):
223
 
224
  input_points = video_state["input_points"]
 
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
  def inference_and_return_video(dilation_iterations, num_inference_steps, video_state=None):
183
  if video_state["origin_images"] is None or video_state["masks"] is None:
184
  return None
 
217
  clip.write_videofile(video_file, codec='libx264', audio=False, verbose=False, logger=None)
218
  return video_file
219
 
 
220
  def track_video(n_frames, video_state):
221
 
222
  input_points = video_state["input_points"]