Spaces:
Paused
Paused
μ μν
commited on
Commit
β’
e35a5a5
1
Parent(s):
f3ee2f7
edit app.py
Browse files
app.py
CHANGED
|
@@ -3,7 +3,7 @@ import PIL.Image as Image
|
|
| 3 |
from ultralytics import YOLO
|
| 4 |
|
| 5 |
|
| 6 |
-
model = YOLO('./best.pt')
|
| 7 |
|
| 8 |
def infer(img_path):
|
| 9 |
results = model(img_path)
|
|
|
|
| 3 |
from ultralytics import YOLO
|
| 4 |
|
| 5 |
|
| 6 |
+
model = YOLO('./models/best.pt')
|
| 7 |
|
| 8 |
def infer(img_path):
|
| 9 |
results = model(img_path)
|