- 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
|