Spaces:

amirgame197
/

Remove-Video-Background

Running on Zero

amirgame197 commited on Feb 22

Commit

5e5187b

•

1 Parent(s): 5d213bc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,10 +9,10 @@ from transparent_background import Remover
 @spaces.GPU()
 def doo(video, mode):
-    if(mode == 'Normal'):
-        remover = Remover()
-    else:
         remover = Remover(mode='fast')
     cap = cv2.VideoCapture(video)
     fps = cap.get(cv2.CAP_PROP_FPS)

 @spaces.GPU()
 def doo(video, mode):
+    if(mode == 'Fast'):
         remover = Remover(mode='fast')
+    else:
+        remover = Remover()
     cap = cv2.VideoCapture(video)
     fps = cap.get(cv2.CAP_PROP_FPS)