amirgame197 commited on
Commit
077d557
1 Parent(s): 5763e52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import gradio as gr
2
  import cv2
3
  import numpy as np
@@ -7,6 +8,7 @@ from transparent_background import Remover
7
 
8
  remover = Remover(mode='fast')
9
 
 
10
  def doo(video):
11
  cap = cv2.VideoCapture(video)
12
  fps = cap.get(cv2.CAP_PROP_FPS)
 
1
+ import spaces
2
  import gradio as gr
3
  import cv2
4
  import numpy as np
 
8
 
9
  remover = Remover(mode='fast')
10
 
11
+ @spaces.GPU
12
  def doo(video):
13
  cap = cv2.VideoCapture(video)
14
  fps = cap.get(cv2.CAP_PROP_FPS)