Stroke-ia commited on
Commit
c1a2296
·
verified ·
1 Parent(s): f76664d

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +3 -3
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 os,cv2
 
9
 
10
  # Load YOLO model
11
- MODEL_PATH = os.path.join(os.path.dirname(__file__), "best.pt")
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 = [