Update app.py

app.py

@@ -7,10 +7,12 @@ import time
 from PIL import Image
 from transparent_background import Remover
 
-remover = Remover(mode='fast')
-
 @spaces.GPU
-def doo(video):
+def doo(video, mode):
+    if(mode == 'Normal'):
+        remover = Remover()
+    if(mode == 'Fast'):
+        remover = Remover(mode='fast')
     cap = cv2.VideoCapture(video)
     fps = cap.get(cv2.CAP_PROP_FPS)
     
@@ -26,7 +28,7 @@ def doo(video):
         if ret is False:
             break
 
-        if time.time() - start_time >= 6:
+        if time.time() - start_time >= 55:
             print("GPU Timeout is coming")
             cap.release()
             writer.release()
@@ -46,5 +48,6 @@ def doo(video):
     writer.release()
     return 'output.mp4'
 
-iface = gr.Interface(fn=doo, inputs="video", outputs="video")
+iface = gr.Interface(fn=doo, [gr.inputs.Video(type='filepath'),
+    gr.inputs.Radio(['Normal','Fast'], type="value", default='Fast', label='Select mode')], outputs="video")
 iface.launch()