Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import torch
 import gradio as gr
 import subprocess
 import datetime
@@ -33,6 +34,8 @@ def check_for_mp4_in_outputs(given_folder):
 
 def infer(input_video, cropped_and_aligned):
 
+    torch.cuda.empty_cache()
+
     filepath = input_video
     # Get the current timestamp
     timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
@@ -44,7 +47,9 @@ def infer(input_video, cropped_and_aligned):
         run_command(f"{sys.executable} inference_keep.py -i={filepath} -o={output_folder_name} --has_aligned --save_video -s=1")
     else:
         run_command(f"{sys.executable} inference_keep.py -i={filepath} -o={output_folder_name} --draw_box --save_video -s=1 --bg_upsampler=realesrgan")
-
+    
+    torch.cuda.empty_cache()
+    
     # Call the function and print the result
     this_infer_folder = os.path.splitext(os.path.basename(filepath))[0]
     joined_path = os.path.join(output_folder_name, this_infer_folder)