Ryan-PR commited on
Commit
f77a896
·
verified ·
1 Parent(s): a1ea72a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -319,7 +319,6 @@ def clear_ref_clicks(ref_state):
319
  return Image.fromarray(ref_state["origin_image"])
320
 
321
 
322
- @torch.no_grad()
323
  @spaces.GPU(duration=40)
324
  def track_video(n_frames, video_state):
325
  input_points = video_state["input_points"]
@@ -408,7 +407,6 @@ def track_video(n_frames, video_state):
408
  return video_file, video_state
409
 
410
 
411
- @torch.no_grad()
412
  @spaces.GPU(duration=150)
413
  def inference_and_return_video(
414
  dilate_radius,
 
319
  return Image.fromarray(ref_state["origin_image"])
320
 
321
 
 
322
  @spaces.GPU(duration=40)
323
  def track_video(n_frames, video_state):
324
  input_points = video_state["input_points"]
 
407
  return video_file, video_state
408
 
409
 
 
410
  @spaces.GPU(duration=150)
411
  def inference_and_return_video(
412
  dilate_radius,