Hugging Face's logo

Spaces:
Hkronboch
/
Trashdetection
Sleeping

App Files Community
5
Hkronboch commited on
Commit
9f24354
·
1 Parent(s): fd85a0e

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('turhancan97/yolov8-segment-trash-detection')
4
  prediction = model.predict(image, imgsz=image_size, show=False, save=False)
 
1
  from ultralytics import YOLO
2
 
3
+ model = YOLO('yolov8m-seg.pt')
4
  prediction = model.predict(image, imgsz=image_size, show=False, save=False)