Update app

app.py

@@ -64,10 +64,11 @@ def preprocess_video(path, result_folder, padding=20):
 	for k in range(len(vr)):
 		all_frames.append(vr[k].asnumpy())
 	all_frames = np.asarray(all_frames)
+	print("Extracted the frames for pre-processing")
 
 	# Load YOLOv5 model (pre-trained on COCO dataset)
 	yolo_model = YOLO("yolov9c.pt")
-
+	print("Loaded the YOLO model")
 
 	if frame_count < 25:
 		msg = "Not enough frames to process! Please give a longer video as input"
@@ -76,7 +77,7 @@ def preprocess_video(path, result_folder, padding=20):
 	person_videos = {}
 	person_tracks = {}
 
-	for frame_idx in range(frame_count):
+	for frame_idx in tqdm(range(frame_count)):
 	
 		frame = all_frames[frame_idx]
 	
@@ -124,6 +125,8 @@ def preprocess_video(path, result_folder, padding=20):
 		msg = "Oops! Could not load the audio file. Please check the input video and try again."
 		return None, None, None, msg
 
+	print("Extracted the audio from the video")
+
 	# For the person detected, crop the frame based on the bounding box
 	if len(person_videos[0]) > frame_count-10:
 		crop_filename = os.path.join(result_folder, "preprocessed_video.avi")