Hugging Face's logo

Spaces:
Stroke-ia
/
API
Sleeping

App Files Community
Stroke-ia commited on
Commit
9363073
·
verified ·
1 Parent(s): 600e7ec

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +1 -1
api.py CHANGED
@@ -5,7 +5,7 @@ from ultralytics import YOLO
5
  import numpy as np
6
  from PIL import Image
7
  import io
8
- import cv2
9
 
10
  # Load YOLO model
11
  MODEL_PATH = os.path.join(os.path.dirname(__file__), "best.pt")
 
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")