Spaces:

Hkronboch
/

Trashdetection

Sleeping

Trashdetection / app.py

Update app.py

9f24354 over 1 year ago

137 Bytes

	from ultralytics import YOLO

	model = YOLO('yolov8m-seg.pt')
	prediction = model.predict(image, imgsz=image_size, show=False, save=False)