Spaces:
Running
on
Zero
Running
on
Zero
roychao19477
commited on
Commit
·
821d472
1
Parent(s):
da43b5a
Upload to debug
Browse files
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)
|