Update app.py

app.py

@@ -2,6 +2,7 @@ import spaces
 import gradio as gr
 import cv2
 import numpy as np
+import gc  # For manual garbage collection
 
 from PIL import Image
 from transparent_background import Remover
@@ -32,9 +33,15 @@ def doo(video):
         print(f"Processing: {processed_frames}")
         out = remover.process(img, type='green')
         writer.write(cv2.cvtColor(np.array(out), cv2.COLOR_BGR2RGB))
+        
+        # Release GPU memory after processing a certain number of frames
+        if processed_frames % 100 == 0:
+            del out  # Delete the variable holding the processed frame
+            gc.collect()  # Perform manual garbage collection to release GPU memory
+    
     cap.release()
     writer.release()
     return 'output.mp4'
 
 iface = gr.Interface(fn=doo, inputs="video", outputs="video")
-iface.launch()
+iface.launch()