tomrb commited on
Commit
015372b
1 Parent(s): e2e0738
Files changed (2) hide show
  1. app.py +1 -1
  2. 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 ultralyticscustom import YOLO
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