Hugging Face's logo

Spaces:
Hkronboch
/
Trashdetection
Sleeping

App Files Community
5
Hkronboch commited on
Commit
318543e
·
1 Parent(s): 8f33028

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,4 +1,4 @@
1
  from ultralytics import YOLO
2
 
3
  model = YOLO('yolov8m-seg.pt')
4
- prediction = model.predict('trash_01.jpg', imgsz=(1000,1333), show=True, save=False)
 
1
  from ultralytics import YOLO
2
 
3
  model = YOLO('yolov8m-seg.pt')
4
+ prediction = model.predict('trash_01.jpg', imgsz=(1000,1333), show=True, save=True)