Spaces:
Sleeping
Sleeping
Commit
•
b66004e
1
Parent(s):
869b645
fix: bug
Browse files- app.py +1 -2
- model_- 19 may 2024 15_13.pt +0 -3
app.py
CHANGED
|
@@ -17,7 +17,6 @@ image_path= [
|
|
| 17 |
]
|
| 18 |
|
| 19 |
# Load YOLO model
|
| 20 |
-
# model = YOLO('model_- 19 may 2024 15_13.pt')
|
| 21 |
model = YOLO('nakamura196/yolov8-ndl-layout')
|
| 22 |
|
| 23 |
def yolov8_img_inference(
|
|
@@ -34,7 +33,7 @@ def yolov8_img_inference(
|
|
| 34 |
Returns:
|
| 35 |
Rendered image
|
| 36 |
"""
|
| 37 |
-
results = model.predict(image, conf=conf_threshold, iou=iou_threshold)
|
| 38 |
render = render_result(model=model, image=image, result=results[0])
|
| 39 |
|
| 40 |
json_data = json.loads(results[0].tojson())
|
|
|
|
| 17 |
]
|
| 18 |
|
| 19 |
# Load YOLO model
|
|
|
|
| 20 |
model = YOLO('nakamura196/yolov8-ndl-layout')
|
| 21 |
|
| 22 |
def yolov8_img_inference(
|
|
|
|
| 33 |
Returns:
|
| 34 |
Rendered image
|
| 35 |
"""
|
| 36 |
+
results = model.predict(image, conf=conf_threshold, iou=iou_threshold, device="cpu")
|
| 37 |
render = render_result(model=model, image=image, result=results[0])
|
| 38 |
|
| 39 |
json_data = json.loads(results[0].tojson())
|
model_- 19 may 2024 15_13.pt
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:8acb10e585c68b31142f2f4470606da52ce054ab467368b6bfe22ca547620e12
|
| 3 |
-
size 136737193
|
|
|
|
|
|
|
|
|
|
|
|