e1250 commited on
Commit
3a8656e
·
1 Parent(s): ac14e44
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -32,8 +32,8 @@ async def lifespan(app: FastAPI):
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")
37
 
38
  safety_detection_path = hf_hub_download(repo_id="e1250/safety_detection", filename="yolo_smoke_fire.pt")
39
  app.state.safety_detection_model = YOLO_Detector(safety_detection_path)
 
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
+ depth_model_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=depth_model_path, DEVICE="cuda")
37
 
38
  safety_detection_path = hf_hub_download(repo_id="e1250/safety_detection", filename="yolo_smoke_fire.pt")
39
  app.state.safety_detection_model = YOLO_Detector(safety_detection_path)