Spaces:
Sleeping
Sleeping
danielhshi8224
commited on
Commit
·
6c3c94e
1
Parent(s):
8fecd1d
update OD model
Browse files- app.py +4 -3
- 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, "
|
| 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 |
-
"
|
| 223 |
-
"
|
|
|
|
| 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
|