Update app.py

app.py

@@ -319,6 +319,7 @@ def clear_ref_clicks(ref_state):
     return Image.fromarray(ref_state["origin_image"])
 
 
+@torch.no_grad()
 @spaces.GPU(duration=40)
 def track_video(n_frames, video_state):
     input_points = video_state["input_points"]
@@ -398,6 +399,7 @@ def track_video(n_frames, video_state):
     return video_file, video_state
 
 
+@torch.no_grad()
 @spaces.GPU(duration=150)
 def inference_and_return_video(
     dilate_radius,