e1250 commited on
Commit
ac14e44
·
1 Parent(s): 85adf6e
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -30,7 +30,7 @@ async def lifespan(app: FastAPI):
30
  asyncio.create_task(log_system_metrics(logger, logger_interval_sec=settings.intervals.system_metrics_seconds))
31
 
32
  # Using this way to can store data. it is acts as a dict which holds instances
33
- app.state.detection_model = YOLO_Detector()
34
 
35
  safety_detection_path = hf_hub_download(repo_id="depth-anything/Depth-Anything-V2-Small", filename="depth_anything_v2_vits.pth")
36
  app.state.depth_model = DepthAnything(encoder=settings.depth.encoder, depth_model_path=settings.depth.model_path, DEVICE="cuda")
 
30
  asyncio.create_task(log_system_metrics(logger, logger_interval_sec=settings.intervals.system_metrics_seconds))
31
 
32
  # Using this way to can store data. it is acts as a dict which holds instances
33
+ app.state.detection_model = YOLO_Detector(model_path="yolo26n.pt")
34
 
35
  safety_detection_path = hf_hub_download(repo_id="depth-anything/Depth-Anything-V2-Small", filename="depth_anything_v2_vits.pth")
36
  app.state.depth_model = DepthAnything(encoder=settings.depth.encoder, depth_model_path=settings.depth.model_path, DEVICE="cuda")