fix: load models from hf and not local

api/routers/camera_stream.py

@@ -67,7 +67,7 @@ async def websocket_detect(
                         pass
                 
                 await frame_queue.put(frame_bytes)
-        except WebScoketDisconnect:
+        except WebSocketDisconnect:
             raise
     
     async def process_frames():

config/config.yaml

@@ -17,11 +17,11 @@ prometheus_port: 9091
 paths:
   # project_dir: &proj_dir G:\MyComputer\Workspace\Projects\gp-tracking-dashboard\tracking_dashboard
   project_dir: &proj_dir .
-  models_dir: &models_dir !join [*proj_dir, ai, dl_models]
+  # models_dir: &models_dir !join [*proj_dir, ai, dl_models]
   # logs_dir:  !join [*proj_dir, backend, config, logs, logs] 
 
 yolo:
-  model_path: !join [*models_dir, yolo26s.pt] 
+  # model_path: !join [*models_dir, yolo26s.pt] 
   classes:
     - person
   batch_size: 16
@@ -31,11 +31,11 @@ yolo:
   data_path: ""
 
 security_detector:
-  model_path: !join [*models_dir, "yolo_smoke_fire.pt"]
+  # model_path: !join [*models_dir, "yolo_smoke_fire.pt"]
   classes:
     - fire
     - smoke
 
 depth: 
-  model_path: !join [*models_dir, depth_anything_v2_vits.pth]  
+  # model_path: !join [*models_dir, depth_anything_v2_vits.pth]  
   encoder: "vits"

main.py

@@ -30,11 +30,13 @@ async def lifespan(app: FastAPI):
     asyncio.create_task(log_system_metrics(logger, logger_interval_sec=settings.intervals.system_metrics_seconds))    
     
     # Using this way to can store data. it is acts as a dict which holds instances
-    app.state.detection_model = YOLO_Detector(settings.yolo.model_path)
+    app.state.detection_model = YOLO_Detector()
+
+    safety_detection_path = hf_hub_download(repo_id="depth-anything/Depth-Anything-V2-Small", filename="depth_anything_v2_vits.pth")
     app.state.depth_model = DepthAnything(encoder=settings.depth.encoder, depth_model_path=settings.depth.model_path, DEVICE="cuda")
 
-    # safety_detection_path = hf_hub_download(repo_id="e1250/safety_detection", filename="yolo_smoke_fire.pt")
-    app.state.safety_detection_model = YOLO_Detector(settings.security_detector.model_path)
+    safety_detection_path = hf_hub_download(repo_id="e1250/safety_detection", filename="yolo_smoke_fire.pt")
+    app.state.safety_detection_model = YOLO_Detector(safety_detection_path)
 
     app.state.logger = logger
     app.state.settings = settings