Spaces:

Hkronboch
/

Trashdetection

Build error

Hkronboch commited on Oct 11, 2023

Commit

2aba432

1 Parent(s): 318543e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,4 +1,4 @@
 from ultralytics import YOLO
 model = YOLO('yolov8m-seg.pt')
-prediction = model.predict('trash_01.jpg', imgsz=(1000,1333), show=True, save=True)

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