Update app.py

app.py

@@ -7,7 +7,7 @@ import time
 from PIL import Image
 from transparent_background import Remover
 
-remover = Remover()
+remover = Remover(mode='fast')
 
 @spaces.GPU
 def doo(video):
@@ -17,13 +17,17 @@ def doo(video):
     writer = None
     
     processed_frames = 0
+    start_time = time.time()
     
     while cap.isOpened():
         ret, frame = cap.read()
     
         if ret is False:
             break
-            
+
+        if time.time() - start_time >= 60:
+            print("GPU Timeout is coming")
+            return 'output.mp4'
         frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) 
         img = Image.fromarray(frame).convert('RGB')