Trashdetection / app.py
Hkronboch's picture
Update app.py
318543e
raw
history blame
145 Bytes
from ultralytics import YOLO
model = YOLO('yolov8m-seg.pt')
prediction = model.predict('trash_01.jpg', imgsz=(1000,1333), show=True, save=True)