Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from PIL import Image, ImageDraw, ImageFont
|
|
4 |
import random
|
5 |
|
6 |
# Load YOLO model (ensure the model file exists in the working directory)
|
7 |
-
YOLO_MODEL_PATH = "
|
8 |
model = YOLO(YOLO_MODEL_PATH, task='detect').to("cpu")
|
9 |
|
10 |
# Define a set of colors for different classes
|
|
|
4 |
import random
|
5 |
|
6 |
# Load YOLO model (ensure the model file exists in the working directory)
|
7 |
+
YOLO_MODEL_PATH = "best-New1Line.pt"
|
8 |
model = YOLO(YOLO_MODEL_PATH, task='detect').to("cpu")
|
9 |
|
10 |
# Define a set of colors for different classes
|