Spaces:
Build error
Build error
Mnjar commited on
Commit ·
d1684c9
1
Parent(s): 7549b99
Fix mode path
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ from PIL import Image, ImageDraw, ImageFont
|
|
| 5 |
from ultralytics import YOLO
|
| 6 |
|
| 7 |
# Load YOLO model
|
| 8 |
-
model = YOLO('
|
| 9 |
|
| 10 |
def predict(image):
|
| 11 |
"""
|
|
|
|
| 5 |
from ultralytics import YOLO
|
| 6 |
|
| 7 |
# Load YOLO model
|
| 8 |
+
model = YOLO('TrashDetection/trash_detection.pt')
|
| 9 |
|
| 10 |
def predict(image):
|
| 11 |
"""
|