e1250 commited on
Commit
7521897
·
1 Parent(s): b0aa005
Files changed (2) hide show
  1. config/config.yaml +0 -3
  2. 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