roychao19477 commited on
Commit
23bc33a
·
1 Parent(s): 385567a

Test on lengths

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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 = []