Update app.py

app.py

@@ -1,12 +1,12 @@
 import spaces
 import gradio as gr
-#import cv2
+import cv2
 #import numpy as np
 
 #from PIL import Image
 #from transparent_background import Remover
 
-import rembg
+from backgroundremover import BackgroundRemover
 
 #remover = Remover()
 
@@ -43,7 +43,24 @@ import rembg
 
 @spaces.GPU
 def doo(video):
-    rembg.greenscreen(video, gpu=True, output="output.mp4", format="mp4", alpha=True, gpubatchsize=10)
+    video = cv2.VideoCapture(video)
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v') 
+    out = cv2.VideoWriter('output.mp4', fourcc, 20.0, (640, 480))
+    while(video.isOpened()):
+        ret, frame = video.read()
+        if ret == True:
+            # Remove the background from the frame
+            no_bg_frame = remover.remove(frame)
+            
+            # Write the frame into the file 'output.mp4'
+            out.write(no_bg_frame)
+        else:
+            break
+    
+    # Release everything when the job is finished
+    video.release()
+    out.release()
+    cv2.destroyAllWindows()
     return 'output.mp4'
 
 iface = gr.Interface(fn=doo, inputs="video", outputs="video")