Spaces:
Sleeping
Sleeping
fix:
Browse files
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 |
-
|
| 36 |
-
app.state.depth_model = DepthAnything(encoder=settings.depth.encoder, depth_model_path=
|
| 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)
|