Update app.py

app.py

@@ -272,7 +272,8 @@ def sam_process(input_first_frame_image, checkpoint, tracking_points, trackings_
     first_frame_output_filename = "output_first_frame.jpg"
     plt.savefig(first_frame_output_filename, format='jpg')
     plt.close()
-
+    torch.cuda.empty_cache()
+    
     return "output_first_frame.jpg", frame_names, inference_state
 
 def propagate_to_all(video_in, checkpoint, stored_inference_state, stored_frame_names, video_frames_dir, vis_frame_type, progress=gr.Progress(track_tqdm=True)):   
@@ -322,6 +323,7 @@ def propagate_to_all(video_in, checkpoint, stored_inference_state, stored_frame_
         # Append the file path to the list
         jpeg_images.append(output_filename)
 
+    torch.cuda.empty_cache()
     
 
     if vis_frame_type == "check":