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 = []
|