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