Spaces:

Hkronboch
/

Trashdetection

Sleeping

Trashdetection / app.py

Update app.py

2aba432 over 1 year ago

159 Bytes

	from ultralytics import YOLO

	model = YOLO('yolov8m-seg.pt')
	prediction = model.predict('trash_01.jpg', imgsz=(1024,1024), show=True, save=True, save_dir=".")