Spaces:

tomrb
/

YOLOv8-TO

Sleeping

tomrb commited on May 1, 2024

Commit

e2e0738

•

1 Parent(s): d943e69

m

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ try:
 except ImportError:
     import os
     os.system('pip install -e ./yolov8-to')
-    from ultralytics import YOLO
 def process_image(image,conf,iou):
     model = YOLO('./trained_models/nano.pt')

 except ImportError:
     import os
     os.system('pip install -e ./yolov8-to')
+    from ultralyticscustom import YOLO
 def process_image(image,conf,iou):
     model = YOLO('./trained_models/nano.pt')