Update app.py

app.py

@@ -11,6 +11,7 @@ import cv2
 import json
 import gradio as gr
 import sys
+import gc 
 
 BASE = os.path.dirname(os.path.abspath(__file__))
 PREPROCESS_DIR = os.path.join(BASE, "wan", "modules", "animate", "preprocess")
@@ -289,6 +290,9 @@ def _animate(input_video, max_duration_s, edited_frame, rc_bool, session_id = No
     total_time = preprocess_time + generate_time
     print(f"Total time: {total_time:.2f} seconds")
 
+    gc.collect()
+    torch.cuda.empty_cache()
+
     return output_video_path
 
 def animate_scene(input_video, max_duration_s, edited_frame, rc_str, session_id = None, progress=gr.Progress(track_tqdm=True),):