Update app.py

app.py

@@ -1,4 +1,4 @@
-import spaces
+#import spaces
 import gradio as gr
 import cv2
 import numpy as np
@@ -7,12 +7,12 @@ import time
 from PIL import Image
 from transparent_background import Remover
 
-@spaces.GPU
+#@spaces.GPU
 def doo(video, mode):
-    #if(mode == 'Normal'):
-    #    remover = Remover()
-    #if(mode == 'Fast'):
-    remover = Remover(mode='fast')
+    if(mode == 'Normal'):
+        remover = Remover()
+    if(mode == 'Fast'):
+        remover = Remover(mode='fast')
     cap = cv2.VideoCapture(video)
     fps = cap.get(cv2.CAP_PROP_FPS)
     
@@ -20,7 +20,7 @@ def doo(video, mode):
     
     processed_frames = 0
     start_time = time.time()
-    print(start_time)
+    print('Starting on mode: ' + mode)
     
     while cap.isOpened():
         ret, frame = cap.read()
@@ -48,5 +48,9 @@ def doo(video, mode):
     writer.release()
     return 'output.mp4'
 
-iface = gr.Interface(fn=doo, inputs="video", outputs="video")
+iface = gr.Interface(
+    fn=doo, 
+    inputs=["video", gr.components.Radio(['Normal', 'Fast'], label='Select mode', value='Normal', info= 'Normal is more accurate, but takes longer. | Fast has lower accuracy so the process will be faster')], 
+    outputs="video"
+)
 iface.launch()