Update app.py

app.py

@@ -10,31 +10,28 @@ remover = Remover()
 
 @spaces.GPU
 def doo(video):
-    try:
-        cap = cv2.VideoCapture(video)
-        fps = cap.get(cv2.CAP_PROP_FPS)
-        
-        processed_frames = 0
-        writer = None
-        
-        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')
+    cap = cv2.VideoCapture(video)
+    fps = cap.get(cv2.CAP_PROP_FPS)
+    
+    processed_frames = 0
+    writer = None
         
-            if writer is None:
-                writer = cv2.VideoWriter('output.mp4', cv2.VideoWriter_fourcc(*'mp4v'), fps, img.size)
+    while cap.isOpened():
+        ret, frame = cap.read()
     
-            processed_frames += 1
-            print(f"Processing: {processed_frames}")
-            out = remover.process(img, type='green')
-            writer.write(cv2.cvtColor(np.array(out), cv2.COLOR_BGR2RGB))
-    except:
-        pass
+        if ret is False:
+            break
+            
+        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)
+
+        processed_frames += 1
+        print(f"Processing: {processed_frames}")
+        out = remover.process(img, type='green')
+        writer.write(cv2.cvtColor(np.array(out), cv2.COLOR_BGR2RGB))
     cap.release()
     writer.release()
     return 'output.mp4'