Update api.py
Browse files
api.py
CHANGED
|
@@ -5,11 +5,11 @@ from ultralytics import YOLO
|
|
| 5 |
import numpy as np
|
| 6 |
from PIL import Image
|
| 7 |
import io
|
| 8 |
-
import
|
|
|
|
| 9 |
|
| 10 |
# Load YOLO model
|
| 11 |
-
|
| 12 |
-
model = YOLO(MODEL_PATH)
|
| 13 |
|
| 14 |
# Class labels
|
| 15 |
CLASS_NAMES = [
|
|
|
|
| 5 |
import numpy as np
|
| 6 |
from PIL import Image
|
| 7 |
import io
|
| 8 |
+
import cv2
|
| 9 |
+
import requests # ✅ ajouté
|
| 10 |
|
| 11 |
# Load YOLO model
|
| 12 |
+
model = YOLO("best.pt")
|
|
|
|
| 13 |
|
| 14 |
# Class labels
|
| 15 |
CLASS_NAMES = [
|