roychao19477 commited on
Commit
821d472
·
1 Parent(s): da43b5a

Upload to debug

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -166,6 +166,7 @@ def extract_resampled_audio(video_path, target_sr=16000):
166
  def yolo_detection(frame, verbose=False):
167
  return model(frame, verbose=verbose)[0]
168
 
 
169
  def extract_faces(video_file):
170
  cap = cv2.VideoCapture(video_file)
171
  fps = cap.get(cv2.CAP_PROP_FPS)
 
166
  def yolo_detection(frame, verbose=False):
167
  return model(frame, verbose=verbose)[0]
168
 
169
+ @spaces.GPU
170
  def extract_faces(video_file):
171
  cap = cv2.VideoCapture(video_file)
172
  fps = cap.get(cv2.CAP_PROP_FPS)