Update app

preprocess/inference_preprocess.py

@@ -166,11 +166,13 @@ def inference_video(opt, padding=0):
 	videofile = os.path.join(opt.avi_dir, 'video.avi')
 	vidObj = cv2.VideoCapture(videofile)
 	yolo_model = YOLO("yolov9t.pt")
-
 	dets = []
 	fidx = 0
+	global dets, fidx
+
 	print("Detecting people in the video using YOLO (slowest step in the pipeline)...")
 	def generate_detections():
+		global dets, fidx
 		while True:
 			success, image = vidObj.read()
 			if not success:
@@ -197,6 +199,8 @@ def inference_video(opt, padding=0):
 			fidx += 1
 			yield
 
+		return dets
+
 	for _ in tqdm(generate_detections()): 
 		pass