Spaces:
Running
Running
fix:
Browse files- config/config.yaml +0 -3
- main.py +0 -1
config/config.yaml
CHANGED
|
@@ -30,6 +30,3 @@ depth:
|
|
| 30 |
model_name: "depth_anything_v2_vits.pth"
|
| 31 |
device: "cpu"
|
| 32 |
encoder: "vits"
|
| 33 |
-
|
| 34 |
-
hf_cache_dir: ".hf_cache"
|
| 35 |
-
dagshub_user_token: ""
|
|
|
|
| 30 |
model_name: "depth_anything_v2_vits.pth"
|
| 31 |
device: "cpu"
|
| 32 |
encoder: "vits"
|
|
|
|
|
|
|
|
|
main.py
CHANGED
|
@@ -59,7 +59,6 @@ async def lifespan(app: FastAPI):
|
|
| 59 |
safety_detection_path = hf_fetch_model(
|
| 60 |
repo_id="e1250/safety_detection",
|
| 61 |
filename=settings.security_detector.model_name,
|
| 62 |
-
cache_dir=settings.hf_cache_dir,
|
| 63 |
)
|
| 64 |
app.state.safety_detection_model = YOLO_Detector(safety_detection_path)
|
| 65 |
|
|
|
|
| 59 |
safety_detection_path = hf_fetch_model(
|
| 60 |
repo_id="e1250/safety_detection",
|
| 61 |
filename=settings.security_detector.model_name,
|
|
|
|
| 62 |
)
|
| 63 |
app.state.safety_detection_model = YOLO_Detector(safety_detection_path)
|
| 64 |
|