Update app.py

app.py

@@ -6,37 +6,42 @@ import numpy as np
 from PIL import Image
 from transparent_background import Remover
 
-remover = Remover()
+#remover = Remover()
 
-@spaces.GPU
-def doo(video):
-    cap = cv2.VideoCapture(video)
-    fps = cap.get(cv2.CAP_PROP_FPS)
-    
-    writer = None
+#@spaces.GPU
+#def doo(video):
+    #cap = cv2.VideoCapture(video)
+    #fps = cap.get(cv2.CAP_PROP_FPS)
     
-    processed_frames = 0
+    #writer = None
     
-    while cap.isOpened():
-        ret, frame = cap.read()
+    #processed_frames = 0
     
-        if ret is False:
-            break
+    #while cap.isOpened():
+    #    ret, frame = cap.read()
+    #
+    #    if ret is False:
+    #        break
             
-        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) 
-        img = Image.fromarray(frame).convert('RGB')
+    #    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) 
+    #    img = Image.fromarray(frame).convert('RGB')
     
-        if writer is None:
-            writer = cv2.VideoWriter('output.mp4', cv2.VideoWriter_fourcc(*'mp4v'), fps, img.size)
+    #    if writer is None:
+    #        writer = cv2.VideoWriter('output.mp4', cv2.VideoWriter_fourcc(*'mp4v'), fps, img.size)
 
-        processed_frames += 1
-        print(f"Processing: {processed_frames}")
+    #    processed_frames += 1
+    #    print(f"Processing: {processed_frames}")
         
-        out = remover.process(img, type='green')
-        writer.write(cv2.cvtColor(np.array(out), cv2.COLOR_BGR2RGB))
+    #    out = remover.process(img, type='green')
+    #    writer.write(cv2.cvtColor(np.array(out), cv2.COLOR_BGR2RGB))
     
-    cap.release()
-    writer.release()
+    #cap.release()
+    #writer.release()
+    #return 'output.mp4'
+
+@spaces.GPU
+def doo(video):
+    rembg.greenscreen(video, gpu=True, output="output.mp4", format="mp4", alpha=True, gpubatchsize=10)
     return 'output.mp4'
 
 iface = gr.Interface(fn=doo, inputs="video", outputs="video")