Update app.py

app.py

@@ -5,16 +5,16 @@ import numpy as np
 from PIL import Image
 from transparent_background import Remover
 
-remover = Remover(mode='fast') # custom setting
+remover = Remover(mode='fast')
 
 def doo(video):
-    cap = cv2.VideoCapture(video) # video reader for input
+    cap = cv2.VideoCapture(video)
     fps = cap.get(cv2.CAP_PROP_FPS)
     
     writer = None
     
     while cap.isOpened():
-        ret, frame = cap.read() # read video
+        ret, frame = cap.read()
     
         if ret is False:
             break
@@ -23,14 +23,14 @@ def doo(video):
         img = Image.fromarray(frame).convert('RGB')
     
         if writer is None:
-            writer = cv2.VideoWriter('output.mp4', cv2.VideoWriter_fourcc(*'mp4v'), fps, img.size) # video writer for output
+            writer = cv2.VideoWriter('output.mp4', cv2.VideoWriter_fourcc(*'mp4v'), fps, img.size)
     
-        out = remover.process(img, type='map') # same as image, except for 'rgba' which is not for video.
+        out = remover.process(img, type='green')
         writer.write(cv2.cvtColor(np.array(out), cv2.COLOR_BGR2RGB))
     
     cap.release()
     writer.release()
-    return 'output.mp4' # returning the file path of the processed video
+    return 'output.mp4'
 
 def display_video(output_video):
     return output_video