danielhshi8224 commited on
Commit
6c3c94e
·
1 Parent(s): 8fecd1d

update OD model

Browse files
Files changed (2) hide show
  1. app.py +4 -3
  2. yolo11_150_best.pt +3 -0
app.py CHANGED
@@ -18,7 +18,7 @@ BASE_DIR = os.path.dirname(os.path.abspath(__file__))
18
  MAX_BATCH = 10
19
 
20
  # Option A: local file baked into Space (easiest if allowed)
21
- YOLO_WEIGHTS = os.path.join(BASE_DIR, "yolo11_best.pt")
22
 
23
  # Option B (optional): pull from a private HF model repo using a Space secret
24
  # Set these env vars in your Space if you want auto-download:
@@ -219,8 +219,9 @@ else:
219
  title="🌊 BenthicAI — Object Detection",
220
  description=(
221
  "Run YOLO object detection on multiple images. "
222
- "Place 'yolo11_best.pt' in the repo root, OR set YOLO_REPO_ID (+ HF_TOKEN if private) "
223
- "to fetch from the Hub."
 
224
  ),
225
  )
226
 
 
18
  MAX_BATCH = 10
19
 
20
  # Option A: local file baked into Space (easiest if allowed)
21
+ YOLO_WEIGHTS = os.path.join(BASE_DIR, "yolo11_150_best.pt")
22
 
23
  # Option B (optional): pull from a private HF model repo using a Space secret
24
  # Set these env vars in your Space if you want auto-download:
 
219
  title="🌊 BenthicAI — Object Detection",
220
  description=(
221
  "Run YOLO object detection on multiple images. "
222
+ "Upload up to 10 images at a time. The model detects various benthic species. "
223
+ "Adjust the confidence and IoU thresholds as needed."
224
+
225
  ),
226
  )
227
 
yolo11_150_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c11a0bb413e9f8ae1e124e52d54a20a0a3753f0d50c118da499cab1d420f3eff
3
+ size 5486746