Spaces:
Sleeping
Sleeping
roychao19477 commited on
Commit ·
23bc33a
1
Parent(s): 385567a
Test on lengths
Browse files
app.py
CHANGED
|
@@ -71,6 +71,7 @@ from decord import VideoReader, cpu
|
|
| 71 |
from model import AVSEModule
|
| 72 |
from config import sampling_rate
|
| 73 |
import spaces
|
|
|
|
| 74 |
|
| 75 |
# Load model once globally
|
| 76 |
#ckpt_path = "ckpts/ep215_0906.oat.ckpt"
|
|
@@ -170,6 +171,7 @@ def yolo_detection(frame, verbose=False):
|
|
| 170 |
|
| 171 |
@spaces.GPU
|
| 172 |
def extract_faces(video_file):
|
|
|
|
| 173 |
cap = cv2.VideoCapture(video_file)
|
| 174 |
fps = cap.get(cv2.CAP_PROP_FPS)
|
| 175 |
frames = []
|
|
|
|
| 71 |
from model import AVSEModule
|
| 72 |
from config import sampling_rate
|
| 73 |
import spaces
|
| 74 |
+
import time
|
| 75 |
|
| 76 |
# Load model once globally
|
| 77 |
#ckpt_path = "ckpts/ep215_0906.oat.ckpt"
|
|
|
|
| 171 |
|
| 172 |
@spaces.GPU
|
| 173 |
def extract_faces(video_file):
|
| 174 |
+
time.sleep(5)
|
| 175 |
cap = cv2.VideoCapture(video_file)
|
| 176 |
fps = cap.get(cv2.CAP_PROP_FPS)
|
| 177 |
frames = []
|