update

app.py

@@ -17,8 +17,9 @@ from utils import load_model, infer_uploaded_image, infer_uploaded_video, infer_
 from download_gd import *
 import os
 
-if not os.path.exists('crowded_human_yolov8s.pt'):
-    download_file_from_google_drive('1qCXBDy3YuxS9bqfRIc--cRLo1L3DAYq2', 'crowded_human_yolov8s.pt')
+if not os.path.exists('./crowded_human_yolov8s.pt'):
+    download_file_from_google_drive('1qCXBDy3YuxS9bqfRIc--cRLo1L3DAYq2', './crowded_human_yolov8s.pt')
+
 
 # setting page layout
 st.set_page_config(

config.py

@@ -32,7 +32,7 @@ SOURCES_LIST = ["Image", "Video", "Webcam"]
 # DL model config
 DETECTION_MODEL_DIR = ROOT / 'models' / 'detection'
 #YOLOv8n = DETECTION_MODEL_DIR / "yolov8n.pt"
-YOLOv8s = 'crowded_human_yolov8s.pt'
+YOLOv8s = './crowded_human_yolov8s.pt'
 #YOLOv8m = DETECTION_MODEL_DIR / "yolov8m.pt"
 #YOLOv8l = DETECTION_MODEL_DIR / "yolov8l.pt"
 #YOLOv8x = DETECTION_MODEL_DIR / "yolov8x.pt"