newtechdevng commited on
Commit
cd37ee9
·
verified ·
1 Parent(s): deb376b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,8 +18,8 @@ app.add_middleware(
18
  )
19
 
20
  # Load YOLO model
21
- HF_REPO_ID = "newtechdevng/construction_detection_v2"
22
- MODEL_FILE = "best_v3_yolo11m.pt"
23
  model_path = hf_hub_download(repo_id=HF_REPO_ID, filename=MODEL_FILE)
24
  model = YOLO(model_path)
25
 
 
18
  )
19
 
20
  # Load YOLO model
21
+ HF_REPO_ID = "newtechdevng/construction_detection_fine_tune"
22
+ MODEL_FILE = "best_v2_finetune.pt"
23
  model_path = hf_hub_download(repo_id=HF_REPO_ID, filename=MODEL_FILE)
24
  model = YOLO(model_path)
25