- app.py +1 -1
- requirements.txt +2 -1
app.py
CHANGED
@@ -10,7 +10,7 @@ try:
|
|
10 |
except ImportError:
|
11 |
import os
|
12 |
os.system('pip install -e ./yolov8-to')
|
13 |
-
from
|
14 |
|
15 |
def process_image(image,conf,iou):
|
16 |
model = YOLO('./trained_models/nano.pt')
|
|
|
10 |
except ImportError:
|
11 |
import os
|
12 |
os.system('pip install -e ./yolov8-to')
|
13 |
+
from ultralytics import YOLO
|
14 |
|
15 |
def process_image(image,conf,iou):
|
16 |
model = YOLO('./trained_models/nano.pt')
|
requirements.txt
CHANGED
@@ -1 +1,2 @@
|
|
1 |
-
torch
|
|
|
|
1 |
+
torch
|
2 |
+
-e ./yolov8-to
|