Trashdetection / app.py
Hkronboch's picture
Update app.py
9701bf4
raw
history blame
190 Bytes
pip install ultralytics
from ultralytics import YOLO
model = YOLO('turhancan97/yolov8-segment-trash-detection')
prediction = model.predict(image, imgsz=image_size, show=False, save=False)